From 348b220908e5f6889a3efc2edbf833e6ab65bf59 Mon Sep 17 00:00:00 2001 From: "Ye (Charlotte) Qi" Date: Tue, 25 Mar 2025 00:56:16 -0700 Subject: [PATCH] num_head -> args.num_head Signed-off-by: Ye (Charlotte) Qi --- hopper/tile_scheduler.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hopper/tile_scheduler.hpp b/hopper/tile_scheduler.hpp index 344a5c03d01..1e4f1420127 100644 --- a/hopper/tile_scheduler.hpp +++ b/hopper/tile_scheduler.hpp @@ -388,7 +388,7 @@ class VarlenDynamicPersistentTileScheduler { // If Split, for the purpose of scheduling, we pretend that instead there are // (args.num_splits * args.num_head) number of heads. assert(args.tile_count_semaphore != nullptr); - assert(num_head < (1 << 16)); // We use the top 16 bits to store num_splits & split_idx + assert(args.num_head < (1 << 16)); // We use the top 16 bits to store num_splits & split_idx assert(!Split || args.num_splits < (1 << 8)); // We use the top 8 bits to store num_splits return {args.num_head, args.num_batch, args.qhead_per_khead, args.seqlen,