chore: import upstream snapshot with attribution

2026-07-13 13:36:25 +08:00
commit 26446540fa
3151 changed files with 974126 additions and 0 deletions
@@ -0,0 +1,68 @@
+/*
+ * 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.
+ */
+/*!
+ * \file test_extern_module.cc
+ * \brief Testing code to be compiled by Relax nn.SourceModule
+ */
+#include <dlpack/dlpack.h>
+#include <tvm/ffi/container/tensor.h>
+#include <tvm/ffi/dtype.h>
+#include <tvm/ffi/function.h>
+
+namespace {
+
+int _scalar_add(DLTensor* a, DLTensor* b, DLTensor* c) {
+  TVM_FFI_ICHECK(a->ndim == 0);
+  TVM_FFI_ICHECK(b->ndim == 0);
+  TVM_FFI_ICHECK(c->ndim == 0);
+  TVM_FFI_ICHECK((a->dtype == DLDataType{kDLFloat, 32, 1}));
+  TVM_FFI_ICHECK((b->dtype == DLDataType{kDLFloat, 32, 1}));
+  TVM_FFI_ICHECK((c->dtype == DLDataType{kDLFloat, 32, 1}));
+  float* a_data = static_cast<float*>(a->data);
+  float* b_data = static_cast<float*>(b->data);
+  float* c_data = static_cast<float*>(c->data);
+  *c_data = *a_data + *b_data;
+  return 0;
+}
+
+int _test_sym(DLTensor* a, DLTensor* b, DLTensor* c) {
+  TVM_FFI_ICHECK(a->ndim == 3);  // [x, y, 1]
+  TVM_FFI_ICHECK(b->ndim == 3);  // [y, z, 5]
+  TVM_FFI_ICHECK(c->ndim == 4);  // [x, y, z, 9]
+  TVM_FFI_ICHECK((a->dtype == DLDataType{kDLFloat, 32, 1}));
+  TVM_FFI_ICHECK((b->dtype == DLDataType{kDLFloat, 32, 1}));
+  TVM_FFI_ICHECK((c->dtype == DLDataType{kDLFloat, 32, 1}));
+  int x = a->shape[0];
+  int y = a->shape[1];
+  int z = b->shape[1];
+  TVM_FFI_ICHECK(a->shape[0] == x);
+  TVM_FFI_ICHECK(a->shape[1] == y);
+  TVM_FFI_ICHECK(a->shape[2] == 1);
+  TVM_FFI_ICHECK(b->shape[0] == y);
+  TVM_FFI_ICHECK(b->shape[1] == z);
+  TVM_FFI_ICHECK(b->shape[2] == 5);
+  TVM_FFI_ICHECK(c->shape[0] == x);
+  TVM_FFI_ICHECK(c->shape[1] == y);
+  TVM_FFI_ICHECK(c->shape[2] == z);
+  TVM_FFI_ICHECK(c->shape[3] == 9);
+  return 0;
+}
+}  // namespace
+TVM_FFI_DLL_EXPORT_TYPED_FUNC(ext_scalar_add, _scalar_add);
+TVM_FFI_DLL_EXPORT_TYPED_FUNC(ext_test_sym, _test_sym);