Merge branch 'tmem-pm' into subpart

NVIDIA · Mar 6, 2025 · a31d9c6 · a31d9c6
2 parents 1763d6b + 16ac5fe
commit a31d9c6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/csrc/device_lower/analysis/tensor_memory.cpp b/csrc/device_lower/analysis/tensor_memory.cpp
@@ -341,6 +341,7 @@ computeTMemLdStDataPath(Fusion* fusion, const TMemAlllocationInfo& allocation) {
     for (auto [i, pt] : enumerate(tid_ptypes)) {
       auto id_it = std::find_if(
           loop_domain.begin(), loop_domain.end(), [pt](IterDomain* id) {
+            NVF_ERROR(id != nullptr);
             return id->getParallelType() == pt;
           });
       if (id_it == loop_domain.end()) {