disable ldmatrix

NVIDIA · Nov 1, 2024 · 523c05b · 523c05b
1 parent 33a1918
commit 523c05b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/csrc/scheduler/hopper_multi_matmul.cpp b/csrc/scheduler/hopper_multi_matmul.cpp
@@ -632,11 +632,10 @@ void HopperMultipleMatmulScheduler::addSetsForCacheReads(
     if (auto ldst = dynamic_cast<LoadStoreOp*>(tv_smem->uses().at(0));
         ldst && tv_smem->uses().size() == 1) {
       tv_r = ldst->out()->as<TensorView>();
-      ldst->setOpType(LoadStoreOpType::LdMatrix);
     } else {
       tv_r = cacheAfter(
           tv_smem,
-          LoadStoreOpType::LdMatrix,
+          LoadStoreOpType::Set,
           CacheOp::Unspecified,
           /*propagate_allocation_domain=*/false);
     }