test

BigNoobWasTaken · BigNoobWasTaken · commit 3277e443825f · 2023-12-28T17:22:53.000+07:00
diff --git a/CPythonTensor/module.c b/CPythonTensor/module.c
@@ -8,10 +8,10 @@ typedef struct
     void* t;
 } PyTensorObject;
 
-static PyObject* Tensor_new(PyTypeObject* type, PyObject* args)
+static int Tensor_init(PyTensorObject* self, PyObject* args)
 {
     PyObject *input_array;
-    PyArg_ParseTuple(args, "|O", &input_array);
+    PyArg_ParseTuple(args, "O", &input_array);
 
     PyArrayObject* np_array = PyArray_FROM_OTF(input_array, NPY_FLOAT, NPY_ARRAY_IN_ARRAY);
     if (np_array == NULL) {
@@ -26,15 +26,23 @@ static PyObject* Tensor_new(PyTypeObject* type, PyObject* args)
         c_dims[i] = dims[i];
     }
 
-    PyTensorObject* self;
-    self = type->tp_alloc(type, 0);
     if (self != NULL) {
         self->t = call_tensor(ndims, c_dims, PyArray_DATA(np_array));
     }
     free(c_dims);
     return self;
 }
 
+static PyObject* Tensor_new(PyTypeObject* type, PyObject* args)
+{
+    PyTensorObject* self;
+    self = type->tp_alloc(type, 0);
+    if (self != NULL) {
+        self->t = 0;
+    }
+    return self;
+}
+
 static void Tensor_dealloc(PyTensorObject* self)
 {
     delete_tensor(self->t);
@@ -59,6 +67,7 @@ static PyTypeObject TensorType =
     .tp_basicsize = sizeof(PyTensorObject),
     .tp_itemsize = 0,
     .tp_flags = Py_TPFLAGS_DEFAULT,
+    .tp_init = Tensor_init,
     .tp_new = Tensor_new,
     .tp_dealloc = Tensor_dealloc,
     .tp_methods = Tensor_methods,
diff --git a/CPythonTensor/py_tensor.cpp b/CPythonTensor/py_tensor.cpp
@@ -14,6 +14,11 @@ void delete_tensor(void* t)
     delete t;
 }
 
+void add_tensor(const void* a, const void* b)
+{
+    return;
+}
+
 const char* to_string(void* t)
 {
     Tensor* t1 = static_cast<Tensor*>(t);

Original file line number	Diff line number	Diff line change
`@@ -14,6 +14,11 @@ void delete_tensor(void* t)`
`14`	`14`	`delete t;`
`15`	`15`	`}`
`16`	`16`
	`17`	`+void add_tensor(const void* a, const void* b)`
	`18`	`+{`
	`19`	`+ return;`
	`20`	`+}`
	`21`	`+`
`17`	`22`	`const char* to_string(void* t)`
`18`	`23`	`{`
`19`	`24`	`Tensor* t1 = static_cast<Tensor*>(t);`