From 09cb4079d5b37ca6ad3d11710300e585e08534ce Mon Sep 17 00:00:00 2001
From: protonu <pbasu@nvidia.com>
Date: Fri, 13 Dec 2024 12:57:38 -0800
Subject: [PATCH] fixing a bug

---
 csrc/scheduler/hopper_multi_matmul.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/csrc/scheduler/hopper_multi_matmul.cpp b/csrc/scheduler/hopper_multi_matmul.cpp
index 98bf7cff37c..91048d3374c 100644
--- a/csrc/scheduler/hopper_multi_matmul.cpp
+++ b/csrc/scheduler/hopper_multi_matmul.cpp
@@ -436,7 +436,6 @@ void HopperMultipleMatmulScheduler::scheduleMmaResults() {
 
 void HopperMultipleMatmulScheduler::scheduleEpilogue() {
   std::vector<TensorView*> cached_tvs;
-  std::vector<TensorView*> c_tvs;
 
   // Propagate to (not including) the splitk output if there is a splitk
   // else this is just mma_results_
@@ -478,7 +477,9 @@ void HopperMultipleMatmulScheduler::scheduleEpilogue() {
       // inner-dim with extent 2.
       // TODO: support vectorization_factor.
       d->axis(-1)->parallelize(ParallelType::Vectorize);
-      scheduler_utils::parallelizeAllLike(d, -1, cached_tvs);
+      if (!cached_tvs.empty()) {
+        scheduler_utils::parallelizeAllLike(d, -1, cached_tvs);
+      }
     }
   } else {
     constexpr int64_t stmatrix_tile_m = 16;