Relay call node definitions. (#396)

* Relay function dahlia impls * ws * imports * Update sqrt name, add a few more. * Update batch_matmul * Prims * Make note of issue 401 * Use std_sqrt * Add softmax data.
calyxir · Feb 19, 2021 · 3fafe12 · 3fafe12
1 parent 9da2ee4
commit 3fafe12
Show file tree

Hide file tree

Showing 65 changed files with 3,434 additions and 46 deletions.
diff --git a/frontends/relay/compiler.py b/frontends/relay/compiler.py
@@ -152,7 +152,10 @@ def emit_futil(program) -> str:
     main, func_defs = visitor.visit(relay_program)
     return '\n'.join((
         Program(
-            imports=[Import("primitives/std.lib")],
+            imports=[
+                Import("primitives/std.lib"),
+                Import("primitives/bitnum/math.futil")
+            ],
             components=[main]
         ).doc(),
         emit_components(func_defs)
@@ -163,4 +166,6 @@ def emit_futil(program) -> str:
     import sys
 
     relay_function = relay.fromtext(sys.stdin.read())
-    print(emit_futil(relay_function))
+    print(
+        emit_futil(relay_function)
+    )