Add back overloads to avoid calling the void* one

albanD · albanD · commit efef1a8a9886 · 2024-05-01T14:45:53.000-04:00
diff --git a/torch/csrc/autograd/utils/wrap_outputs.h b/torch/csrc/autograd/utils/wrap_outputs.h
@@ -52,10 +52,18 @@ inline PyObject* wrap(void* value) {
   return THPUtils_packInt64(reinterpret_cast<intptr_t>(value));
 }
 
+inline PyObject* wrap(THPDtype* dtype) {
+  return Py_NewRef(dtype);
+}
+
 inline PyObject* wrap(at::ScalarType scalarType) {
   return Py_NewRef(getTHPDtype(scalarType));
 }
 
+inline PyObject* wrap(THPLayout* layout) {
+  return Py_NewRef(layout);
+}
+
 inline PyObject* wrap(at::Layout layout) {
   return Py_NewRef(getTHPLayout(layout));
 }

Original file line number	Diff line number	Diff line change
`@@ -52,10 +52,18 @@ inline PyObject* wrap(void* value) {`
`52`	`52`	`return THPUtils_packInt64(reinterpret_cast<intptr_t>(value));`
`53`	`53`	`}`
`54`	`54`
	`55`	`+inline PyObject* wrap(THPDtype* dtype) {`
	`56`	`+ return Py_NewRef(dtype);`
	`57`	`+}`
	`58`	`+`
`55`	`59`	`inline PyObject* wrap(at::ScalarType scalarType) {`
`56`	`60`	`return Py_NewRef(getTHPDtype(scalarType));`
`57`	`61`	`}`
`58`	`62`
	`63`	`+inline PyObject* wrap(THPLayout* layout) {`
	`64`	`+ return Py_NewRef(layout);`
	`65`	`+}`
	`66`	`+`
`59`	`67`	`inline PyObject* wrap(at::Layout layout) {`
`60`	`68`	`return Py_NewRef(getTHPLayout(layout));`
`61`	`69`	`}`