[TEST] Add comprehensive ORC JIT tests for error handling, type variety, and cross-library linking

cyx-6 · claude · cyx-6 · commit b5c1f4da5f30 · 2026-03-20T13:29:05.000Z
- Group 1 (pure Python): test empty library, invalid object file path/content,
  independent sessions, library name collision
- Group 2 (type variety): zero-arg, four-arg, float multiply, void return —
  parametrized over C and C++ with new test_types.c/cc source files
- Group 3 (advanced): cross-library symbol resolution via set_link_order with
  new link_order_base/caller source files, error propagation via
  TVMFFIErrorSetRaisedFromCStr (C) and std::runtime_error (C++)

Co-Authored-By: Claude Opus 4.6 &lt;noreply@anthropic.com&gt;
diff --git a/addons/tvm-ffi-orcjit/tests/CMakeLists.txt b/addons/tvm-ffi-orcjit/tests/CMakeLists.txt
@@ -69,6 +69,10 @@ if(NOT WIN32)
   add_test_object(sources/cc/test_funcs_conflict.cc)
   add_test_object(sources/cc/test_ctor_dtor.cc)
   add_test_object(sources/cc/test_call_global.cc)
+  add_test_object(sources/cc/test_types.cc)
+  add_test_object(sources/cc/test_link_order_base.cc)
+  add_test_object(sources/cc/test_link_order_caller.cc)
+  add_test_object(sources/cc/test_error.cc)
 endif()
 
 # Pure C object files — built on all platforms (no C++ runtime deps)
@@ -77,6 +81,10 @@ add_test_object(sources/c/test_funcs.c)
 add_test_object(sources/c/test_funcs2.c)
 add_test_object(sources/c/test_funcs_conflict.c)
 add_test_object(sources/c/test_call_global.c)
+add_test_object(sources/c/test_types.c)
+add_test_object(sources/c/test_link_order_base.c)
+add_test_object(sources/c/test_link_order_caller.c)
+add_test_object(sources/c/test_error.c)
 
 # CUDA object files — optional
 find_package(CUDAToolkit)
diff --git a/addons/tvm-ffi-orcjit/tests/sources/c/test_error.c b/addons/tvm-ffi-orcjit/tests/sources/c/test_error.c
@@ -0,0 +1,31 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+/*
+ * Error propagation test: JIT function signals error via
+ * TVMFFIErrorSetRaisedFromCStr, which should surface as a Python exception.
+ */
+#include <tvm/ffi/c_api.h>
+
+/* test_error: always raises a ValueError */
+TVM_FFI_DLL_EXPORT int __tvm_ffi_test_error(
+    void* self, const TVMFFIAny* args, int32_t num_args, TVMFFIAny* result) {
+  TVMFFIErrorSetRaisedFromCStr("ValueError", "test error");
+  return -1;
+}
diff --git a/addons/tvm-ffi-orcjit/tests/sources/c/test_link_order_base.c b/addons/tvm-ffi-orcjit/tests/sources/c/test_link_order_base.c
@@ -0,0 +1,33 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+/*
+ * Base library for cross-library linking test.
+ * Exports helper_add which is called by test_link_order_caller.c.
+ */
+#include <tvm/ffi/c_api.h>
+
+/* helper_add: add two integers — used as a dependency by the caller library */
+TVM_FFI_DLL_EXPORT int __tvm_ffi_helper_add(
+    void* self, const TVMFFIAny* args, int32_t num_args, TVMFFIAny* result) {
+  result->type_index = kTVMFFIInt;
+  result->zero_padding = 0;
+  result->v_int64 = args[0].v_int64 + args[1].v_int64;
+  return 0;
+}
diff --git a/addons/tvm-ffi-orcjit/tests/sources/c/test_link_order_caller.c b/addons/tvm-ffi-orcjit/tests/sources/c/test_link_order_caller.c
@@ -0,0 +1,35 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+/*
+ * Caller library for cross-library linking test.
+ * References __tvm_ffi_helper_add from test_link_order_base.c via extern
+ * declaration, and exports cross_lib_add which forwards to it.
+ */
+#include <tvm/ffi/c_api.h>
+
+/* Declare external symbol from the base library */
+extern int __tvm_ffi_helper_add(
+    void* self, const TVMFFIAny* args, int32_t num_args, TVMFFIAny* result);
+
+/* cross_lib_add: forwards to helper_add in the base library */
+TVM_FFI_DLL_EXPORT int __tvm_ffi_cross_lib_add(
+    void* self, const TVMFFIAny* args, int32_t num_args, TVMFFIAny* result) {
+  return __tvm_ffi_helper_add(self, args, num_args, result);
+}
diff --git a/addons/tvm-ffi-orcjit/tests/sources/c/test_types.c b/addons/tvm-ffi-orcjit/tests/sources/c/test_types.c
@@ -0,0 +1,60 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+/*
+ * Pure C test functions exercising type variety: zero-arg, four-arg,
+ * float, and void return types via the TVMFFISafeCallType ABI.
+ */
+#include <tvm/ffi/c_api.h>
+
+/* test_zero_arg: ignores arguments, returns integer 42 */
+TVM_FFI_DLL_EXPORT int __tvm_ffi_test_zero_arg(
+    void* self, const TVMFFIAny* args, int32_t num_args, TVMFFIAny* result) {
+  result->type_index = kTVMFFIInt;
+  result->zero_padding = 0;
+  result->v_int64 = 42;
+  return 0;
+}
+
+/* test_four_args: sum of four integer arguments */
+TVM_FFI_DLL_EXPORT int __tvm_ffi_test_four_args(
+    void* self, const TVMFFIAny* args, int32_t num_args, TVMFFIAny* result) {
+  result->type_index = kTVMFFIInt;
+  result->zero_padding = 0;
+  result->v_int64 = args[0].v_int64 + args[1].v_int64 + args[2].v_int64 + args[3].v_int64;
+  return 0;
+}
+
+/* test_float_multiply: multiply two doubles */
+TVM_FFI_DLL_EXPORT int __tvm_ffi_test_float_multiply(
+    void* self, const TVMFFIAny* args, int32_t num_args, TVMFFIAny* result) {
+  result->type_index = kTVMFFIFloat;
+  result->zero_padding = 0;
+  result->v_float64 = args[0].v_float64 * args[1].v_float64;
+  return 0;
+}
+
+/* test_void_function: does nothing, returns None */
+TVM_FFI_DLL_EXPORT int __tvm_ffi_test_void_function(
+    void* self, const TVMFFIAny* args, int32_t num_args, TVMFFIAny* result) {
+  result->type_index = kTVMFFINone;
+  result->zero_padding = 0;
+  result->v_int64 = 0;
+  return 0;
+}
diff --git a/addons/tvm-ffi-orcjit/tests/sources/cc/test_error.cc b/addons/tvm-ffi-orcjit/tests/sources/cc/test_error.cc
@@ -0,0 +1,27 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+// Error propagation test (C++ version): throws std::runtime_error
+// which TVM_FFI_SAFE_CALL_END catches and converts to an InternalError.
+
+#include <tvm/ffi/function.h>
+
+#include <stdexcept>
+
+TVM_FFI_DLL_EXPORT_TYPED_FUNC(test_error, []() -> int {
+  throw std::runtime_error("test error");
+});
diff --git a/addons/tvm-ffi-orcjit/tests/sources/cc/test_link_order_base.cc b/addons/tvm-ffi-orcjit/tests/sources/cc/test_link_order_base.cc
@@ -0,0 +1,24 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+// Base library for cross-library linking test (C++ version).
+// Exports helper_add which is called by test_link_order_caller.cc.
+
+#include <tvm/ffi/function.h>
+
+int helper_add_impl(int a, int b) { return a + b; }
+TVM_FFI_DLL_EXPORT_TYPED_FUNC(helper_add, helper_add_impl);
diff --git a/addons/tvm-ffi-orcjit/tests/sources/cc/test_link_order_caller.cc b/addons/tvm-ffi-orcjit/tests/sources/cc/test_link_order_caller.cc
@@ -0,0 +1,43 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+// Caller library for cross-library linking test (C++ version).
+// References __tvm_ffi_helper_add from test_link_order_base.cc and
+// exports cross_lib_add which forwards to it.
+
+#include <tvm/ffi/c_api.h>
+#include <tvm/ffi/function.h>
+
+extern "C" int __tvm_ffi_helper_add(
+    void* self, const TVMFFIAny* args, int32_t num_args, TVMFFIAny* result);
+
+int cross_lib_add_impl(int a, int b) {
+  TVMFFIAny call_args[2];
+  TVMFFIAny call_result;
+  call_args[0].type_index = kTVMFFIInt;
+  call_args[0].zero_padding = 0;
+  call_args[0].v_int64 = a;
+  call_args[1].type_index = kTVMFFIInt;
+  call_args[1].zero_padding = 0;
+  call_args[1].v_int64 = b;
+  call_result.type_index = kTVMFFINone;
+  call_result.zero_padding = 0;
+  call_result.v_int64 = 0;
+  __tvm_ffi_helper_add(nullptr, call_args, 2, &call_result);
+  return static_cast<int>(call_result.v_int64);
+}
+TVM_FFI_DLL_EXPORT_TYPED_FUNC(cross_lib_add, cross_lib_add_impl);
diff --git a/addons/tvm-ffi-orcjit/tests/sources/cc/test_types.cc b/addons/tvm-ffi-orcjit/tests/sources/cc/test_types.cc
@@ -0,0 +1,26 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+// C++ test functions exercising type variety: zero-arg, four-arg,
+// float, and void return types.
+
+#include <tvm/ffi/function.h>
+
+TVM_FFI_DLL_EXPORT_TYPED_FUNC(test_zero_arg, []() { return 42; });
+TVM_FFI_DLL_EXPORT_TYPED_FUNC(test_four_args, [](int a, int b, int c, int d) { return a + b + c + d; });
+TVM_FFI_DLL_EXPORT_TYPED_FUNC(test_float_multiply, [](double a, double b) { return a * b; });
+TVM_FFI_DLL_EXPORT_TYPED_FUNC(test_void_function, []() {});
diff --git a/addons/tvm-ffi-orcjit/tests/test_basic.py b/addons/tvm-ffi-orcjit/tests/test_basic.py
