debugging mem2reg

Learning-Ray-Tracing-From-Zero · Feb 10, 2025 · 71a9998 · 71a9998
1 parent bd32ded
commit 71a9998
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 16 deletions.
diff --git a/src/xir/passes/dom_tree.cpp b/src/xir/passes/dom_tree.cpp
@@ -4,7 +4,7 @@
 namespace luisa::compute::xir {
 
 inline DomTreeNode::DomTreeNode(BasicBlock *block) noexcept
-    : _parent{nullptr}, _block{block} {}
+    : _block{block}, _parent{nullptr} {}
 
 inline void DomTreeNode::add_child(DomTreeNode *child) noexcept {
     LUISA_DEBUG_ASSERT(child != nullptr && child->_parent == nullptr && child != this, "Invalid child.");

diff --git a/src/xir/tests/test_mem2reg.cpp b/src/xir/tests/test_mem2reg.cpp
@@ -11,22 +11,20 @@ int main(int argc, char *argv[]) {
 
     using namespace luisa::compute;
     auto shader = device.compile<1>([&](UInt n) noexcept {
-        auto x = def<uint>();
-        auto zero = def(0u);
-        auto one = def(1u);
-        $while (true) {
-            $if (n == 0u) {
-                x = zero;
-            }
-            $else {
-                n -= 1u;
-                x = one;
-            };
-            $if (n == 0u) {
-                $break;
-            };
+        auto b = compute::detail::FunctionBuilder::current();
+        auto t = Type::of<uint>();
+        auto x = b->local(t);
+        auto zero = b->literal(t, 0u);
+        auto one = b->literal(t, 1u);
+        $if (n == 0u) {
+        }
+        $elif (n == 1u) {
+            // b->assign(x, one);
+        }
+        $else {
+            b->assign(x, zero);
         };
-        buffer->write(0u, x);
+        buffer->write(0u, def<uint>(x));
     });
 
     uint result = 0u;