From 7fa7d80a1c90fa37bbdc26f3cddddd89268a6d43 Mon Sep 17 00:00:00 2001
From: Dmitry Sidorov <dmitrii.s.sidorov@gmail.com>
Date: Tue, 19 Aug 2025 00:24:49 +0200
Subject: [PATCH 1/2] Reorder lowering passes for LLVM IR generation

If fixes error: Dialect memref' not found for custom op 'memref.subview'
for CNN example.
---
 backend/server.py               | 11 ++++---
 tests/test_torch_llvm_ir_cnn.py | 53 +++++++++++++++++++++++++++++++++
 2 files changed, 60 insertions(+), 4 deletions(-)
 create mode 100644 tests/test_torch_llvm_ir_cnn.py

diff --git a/backend/server.py b/backend/server.py
index ef67041..1a26c7f 100644
--- a/backend/server.py
+++ b/backend/server.py
@@ -450,16 +450,19 @@ def lower_to_llvm_mlir(model, example_input) -> str:
 
     cmd = [
         os.path.join(LLVM_BIN_PATH, "mlir-opt"),
-        '--one-shot-bufferize="bufferize-function-boundaries"',
+        '--one-shot-bufferize=bufferize-function-boundaries',
+        "-canonicalize", "-cse",
         "-convert-linalg-to-loops",
-        "-convert-scf-to-cf",
-        "-convert-cf-to-llvm",
         "-lower-affine",
-        "-finalize-memref-to-llvm",
+        "-convert-scf-to-cf",
+        "-expand-strided-metadata",
         "-convert-math-to-llvm",
         "-convert-arith-to-llvm",
+        "-finalize-memref-to-llvm",
         "-convert-func-to-llvm",
+        "-convert-cf-to-llvm",
         "-reconcile-unrealized-casts",
+        "-canonicalize", "-cse",
         input_path,
     ]
 
diff --git a/tests/test_torch_llvm_ir_cnn.py b/tests/test_torch_llvm_ir_cnn.py
new file mode 100644
index 0000000..7a85fdc
--- /dev/null
+++ b/tests/test_torch_llvm_ir_cnn.py
@@ -0,0 +1,53 @@
+import os
+import pytest
+import httpx
+
+API_URL = os.environ.get("API_URL", "http://localhost:8000/generate_ir")
+
+
+def test_torch_cnn():
+    code = """
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
+
+class SimpleCNN(nn.Module):
+    def __init__(self, num_classes=10):
+        super().__init__()
+        self.conv1 = nn.Conv2d(1, 16, kernel_size=3, stride=1, padding=1)
+        self.conv2 = nn.Conv2d(16, 32, kernel_size=3, stride=1, padding=1)
+        self.fc = nn.Linear(32 * 7 * 7, num_classes)
+
+    def forward(self, x):
+        x = F.relu(self.conv1(x))
+        x = F.max_pool2d(x, 2)
+        x = F.relu(self.conv2(x))
+        x = F.max_pool2d(x, 2)
+        x = torch.flatten(x, 1)
+        x = self.fc(x)
+        return x
+
+model = SimpleCNN(num_classes=10)
+x = torch.randn(8, 1, 28, 28)
+
+"""
+
+    payload = {
+        "code": code,
+        "ir_type": "llvm_ir",
+        "custom_pipeline": [],
+        "torch_mlir_opt": "",
+        "mlir_opt": "",
+        "mlir_translate": "",
+        "llvm_opt": "",
+        "llc": "",
+        "user_tool": "",
+        "dump_after_each_opt": False,
+    }
+
+    response = httpx.post(API_URL, json=payload)
+    assert response.status_code == 200
+
+    ir = response.json()["output"]
+
+    assert "@main" in ir

From 696ccff88bc2c1f8ef7598d8d6a07e308a374456 Mon Sep 17 00:00:00 2001
From: Dmitry Sidorov <dmitrii.s.sidorov@gmail.com>
Date: Tue, 19 Aug 2025 00:27:10 +0200
Subject: [PATCH 2/2] format

---
 backend/server.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/backend/server.py b/backend/server.py
index 1a26c7f..b83a4d1 100644
--- a/backend/server.py
+++ b/backend/server.py
@@ -450,8 +450,9 @@ def lower_to_llvm_mlir(model, example_input) -> str:
 
     cmd = [
         os.path.join(LLVM_BIN_PATH, "mlir-opt"),
-        '--one-shot-bufferize=bufferize-function-boundaries',
-        "-canonicalize", "-cse",
+        "--one-shot-bufferize=bufferize-function-boundaries",
+        "-canonicalize",
+        "-cse",
         "-convert-linalg-to-loops",
         "-lower-affine",
         "-convert-scf-to-cf",
@@ -462,7 +463,8 @@ def lower_to_llvm_mlir(model, example_input) -> str:
         "-convert-func-to-llvm",
         "-convert-cf-to-llvm",
         "-reconcile-unrealized-casts",
-        "-canonicalize", "-cse",
+        "-canonicalize",
+        "-cse",
         input_path,
     ]