fix(core): Fix DLPack Interface (#47)

Author: potatomashed

cpp/registry.h

@@ -659,7 +659,7 @@ inline TypeTable *TypeTable::New() {
   self->SetFunc("mlc.core.TensorFromBytes", Func(::mlc::registry::TensorFromBytes).get());
   self->SetFunc("mlc.core.TensorToBase64", Func(::mlc::registry::TensorToBase64).get());
   self->SetFunc("mlc.core.TensorFromBase64", Func(::mlc::registry::TensorFromBase64).get());
-  self->SetFunc("mlc.core.TensorToDLPack", Func([](TensorObj *tensor) -> void * { return tensor->DLPack(); }).get());
+  self->SetFunc("mlc.core.TensorToDLPack", Func([](Tensor tensor) -> void * { return tensor->DLPack(); }).get());
   self->SetFunc("mlc.printer.DocToPythonScript", Func(::mlc::registry::DocToPythonScript).get());
   self->SetFunc("mlc.printer.ToPython", Func(::mlc::printer::ToPython).get());
 

python/mlc/core/tensor.py

@@ -67,7 +67,7 @@ def base64(self) -> str:
     def from_base64(base64: str) -> Tensor:
         return TensorFromBase64(base64)
 
-    def __dlpack__(self) -> Any:
+    def __dlpack__(self, stream: Any = None) -> Any:
         return tensor_to_dlpack(self)
 
     def __dlpack_device__(self) -> tuple[int, int]: