[mpact][compiler] enable norm and scale test with MPACT JIT (#56)
Note:
We still have an empty first row bug lingering
but this PR at least enables the JITting of the
two kernels with MPACT
diff --git a/test/python/norm.py b/test/python/norm.py
index f589e46..27a0848 100644
--- a/test/python/norm.py
+++ b/test/python/norm.py
@@ -28,7 +28,18 @@
# CHECK: [0.0000, 0.0000, 0.0000, 0.0000, 0.0000, 1.0000, 0.0000, 0.0000],
# CHECK: [0.0000, 0.0000, 0.0000, 0.0000, 0.0000, 0.0000, 1.0000, 0.0000],
# CHECK: [0.0000, 0.0000, 0.0000, 0.0000, 0.0000, 0.0000, 0.0000, 1.0000]{{\]}})
+#
+# TODO: first row?
+#
# CHECK: mpact
+# CHECK: {{\[}}[0. 0. 0. 0. 0. 0. 0. 0. ]
+# CHECK: [0. 0.25 0. 0. 0.25 0. 0. 0. ]
+# CHECK: [0. 0. 1. 0. 0. 0. 0. 0. ]
+# CHECK: [0. 0. 0. 0.25 0.25 0. 0. 0. ]
+# CHECK: [0. 0. 0. 0.25 0.25 0. 0. 0. ]
+# CHECK: [0. 0. 0. 0. 0. 1. 0. 0. ]
+# CHECK: [0. 0. 0. 0. 0. 0. 1. 0. ]
+# CHECK: [0. 0. 0. 0. 0. 0. 0. 1. ]{{\]}}
#
# Run it with PyTorch.
@@ -37,9 +48,6 @@
print(res)
# Run it with MPACT.
-#
-# TODO: make this work, crashes in TORCH-MLIR
-#
print("mpact")
-# res = mpact_jit(net, adj_mat)
-# print(res)
+res = mpact_jit(net, adj_mat)
+print(res)
diff --git a/test/python/scale.py b/test/python/scale.py
index 880fdf2..f8a4443 100644
--- a/test/python/scale.py
+++ b/test/python/scale.py
@@ -22,7 +22,24 @@
# CHECK: [0.0774, 0.1561, 0.1275, 0.3896, 0.0735, 0.1128, 0.0630],
# CHECK: [0.0093, 0.0611, 0.2731, 0.2124, 0.2180, 0.1546, 0.0716],
# CHECK: [0.2026, 0.0115, 0.0481, 0.0839, 0.2826, 0.2749, 0.0964]{{\]}})
+#
+# TODO: first row?
+#
# CHECK: mpact
+# CHECK: {{\[}}[0. 0. 0. 0. 0. 0.
+# CHECK: 0. ]
+# CHECK: [0.30635384 0.15570773 0.21608633 0.11923195 0.13728413 0.00762967
+# CHECK: 0.05770639]
+# CHECK: [0.08555716 0.15095268 0.20310582 0.23290026 0.04687909 0.08217437
+# CHECK: 0.19843069]
+# CHECK: [0.22065267 0.09574053 0.2107584 0.10111907 0.13330552 0.22970453
+# CHECK: 0.00871931]
+# CHECK: [0.07743214 0.15609969 0.12754099 0.3896042 0.07353575 0.11279855
+# CHECK: 0.06298868]
+# CHECK: [0.00931544 0.06112389 0.2730649 0.2123639 0.21801054 0.15456341
+# CHECK: 0.07155795]
+# CHECK: [0.20259099 0.01148908 0.04807246 0.08394676 0.28260148 0.2748705
+# CHECK: 0.09642864]]
#
# Run it with PyTorch.
@@ -31,9 +48,6 @@
print(res)
# Run it with MPACT.
-#
-# TODO: make this work, crashes in TORCH-MLIR
-#
print("mpact")
-# res = mpact_jit(net, features)
-# print(res)
+res = mpact_jit(net, features)
+print(res)
