Fix builds for TVM on Windows

bytedeco · Nov 11, 2023 · 0fbf403 · 0fbf403
1 parent 8f2d975
commit 0fbf403
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 74 deletions.
diff --git a/tvm/src/gen/java/org/bytedeco/tvm/WrappedPythonObject.java b/tvm/src/gen/java/org/bytedeco/tvm/WrappedPythonObject.java
diff --git a/tvm/src/gen/java/org/bytedeco/tvm/global/tvm_runtime.java b/tvm/src/gen/java/org/bytedeco/tvm/global/tvm_runtime.java
@@ -3003,7 +3003,7 @@ public static native int TVMObjectDerivedFrom(@Cast("uint32_t") int child_type_i
  */
 @Namespace("tvm::runtime") public static native @Cast("const char*") BytePointer ArgTypeCode2Str(int type_code);
 
-@Namespace("tvm::runtime") public static native @Cast("std::ostream*") @ByRef @Name("operator <<") Pointer shiftLeft(@Cast("std::ostream*") @ByRef Pointer os, @ByVal DLDevice dev);  // NOLINT(*)
+  // NOLINT(*)
 
 // macro to check type code.
 // #define TVM_CHECK_TYPE_CODE(CODE, T)
@@ -3341,9 +3341,13 @@ public static native int TVMObjectDerivedFrom(@Cast("uint32_t") int child_type_i
  *         indicate an error happens, otherwise it returns normally.
  */
 @Namespace("tvm::runtime") public static native void EnvCheckSignals();
-// Targeting ../WrappedPythonObject.java
-
 
+/** \brief A class that wraps a Python object and preserves its ownership.
+ <p>
+ * This class is used to wrap a PyObject* from the Python API and preserve its ownership.
+ * Allows for the creation of strong references to Python objects, which prevent them from being
+ * garbage-collected as long as the wrapper object exists.
+ */
 // Targeting ../Registry.java
 
 

diff --git a/tvm/src/main/java/org/bytedeco/tvm/presets/tvm_runtime.java b/tvm/src/main/java/org/bytedeco/tvm/presets/tvm_runtime.java
@@ -166,7 +166,9 @@ public void map(InfoMap infoMap) {
                .put(new Info("tvm::runtime::ObjectPtr<tvm::runtime::Object>").pointerTypes("ObjectPtr"))
                .put(new Info("tvm::runtime::ObjAllocatorBase<tvm::runtime::SimpleObjAllocator>",
                              "tvm::runtime::ObjAllocatorBase<SimpleObjAllocator>").pointerTypes("SimpleObjAllocatorBase"))
-               .put(new Info("tvm::runtime::ObjAllocatorBase<tvm::runtime::SimpleObjAllocator>::make_object").skip())
+               .put(new Info("tvm::runtime::ObjAllocatorBase<tvm::runtime::SimpleObjAllocator>::make_object",
+                             "tvm::runtime::WrappedPythonObject",
+                             "tvm::runtime::operator <<(std::ostream&, DLDevice)").skip())
 
                .put(new Info("tvm::runtime::ADTObj").pointerTypes("ADTObj"))
                .put(new Info("tvm::runtime::InplaceArrayBase<tvm::runtime::ADTObj,tvm::runtime::ObjectRef>",