Updated test lowering of float64

Author: Lunderberg

tests/python/unittest/test_target_codegen_llvm.py

@@ -28,7 +28,7 @@
 from tvm import te
 from tvm.contrib import clang, utils
 from tvm.relay.backend import Runtime
-from tvm.script import tir as T
+from tvm.script import tir as T, ir as I
 from tvm.target.codegen import llvm_get_intrinsic_name, llvm_lookup_intrinsic_id
 
 
@@ -1007,19 +1007,25 @@ def test_debug_symbol_for_float64():
     """Check that LLVM can define DWARF debug type for float64
 
     In previous versions, only specific data types could exist in the
-    function signature.  In this test, the "calling_conv" attribute
-    prevents lowering to the PackedFunc API.
+    function signature.  In this test, the subroutine is a context
+    that isn't lowered to the PackedFunc API.
     """
 
-    @T.prim_func
-    def func(a: T.handle("float64"), b: T.handle("float64"), n: T.int64):
-        T.func_attr({"calling_conv": 2})
-        A = T.Buffer(16, "float64", data=a)
-        B = T.Buffer(16, "float64", data=b)
-        for i in range(n):
-            B[i] = A[i]
-
-    tvm.build(func, target="llvm")
+    @I.ir_module
+    class mod:
+        @T.prim_func
+        def main(A: T.Buffer(16, "float64"), B: T.Buffer(16, "float64")):
+            T.func_attr({"global_symbol": "main"})
+            mod.subroutine(A.data, B.data, T.int64(16))
+
+        @T.prim_func
+        def subroutine(a: T.handle("float64"), b: T.handle("float64"), n: T.int64):
+            A = T.Buffer(16, "float64", data=a)
+            B = T.Buffer(16, "float64", data=b)
+            for i in range(n):
+                B[i] = A[i]
+
+    tvm.build(mod, target="llvm")
 
 
 if __name__ == "__main__":