Use the default destructor for PyCapsule

pabloferz · pabloferz · commit b3f69e3237d6 · 2023-08-29T16:35:03.000-05:00
diff --git a/python/PyDLExt.h b/python/PyDLExt.h
@@ -28,7 +28,7 @@ static std::vector<PyTensorBundle> kPyCapsulesPool;
 
 inline PyCapsule enpycapsulate(DLManagedTensorPtr tensor, bool autodestruct = true)
 {
-    auto capsule = PyCapsule(tensor, kDLTensorCapsuleName, nullptr);
+    auto capsule = PyCapsule(tensor, kDLTensorCapsuleName);  // default destructor is nullptr
     if (autodestruct)
         PyCapsule_SetDestructor(
             capsule.ptr(),

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ static std::vector<PyTensorBundle> kPyCapsulesPool;`
`28`	`28`
`29`	`29`	`inline PyCapsule enpycapsulate(DLManagedTensorPtr tensor, bool autodestruct = true)`
`30`	`30`	`{`
`31`		`- auto capsule = PyCapsule(tensor, kDLTensorCapsuleName, nullptr);`
	`31`	`+ auto capsule = PyCapsule(tensor, kDLTensorCapsuleName); // default destructor is nullptr`
`32`	`32`	`if (autodestruct)`
`33`	`33`	`PyCapsule_SetDestructor(`
`34`	`34`	`capsule.ptr(),`