another way to calc size of buf item size, fixed the issue with mssd.…

…tfile (#547)
OAID · Feb 10, 2021 · 9a0bcd4 · 9a0bcd4
1 parent 2b85249
commit 9a0bcd4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytengine/tengine/tensor.py b/pytengine/tengine/tensor.py
@@ -121,7 +121,7 @@ def buf(self, value):
         :return: None
         """
         _LIB.set_tensor_buffer.argtypes = [ctypes.c_void_p, ctypes.c_void_p, ctypes.c_int]
-        value_bytes = _LIB.data_type_size(_LIB.get_tensor_data_type(ctypes.c_void_p(self.tensor)))
+        value_bytes = ctypes.sizeof(Tengine_ctype[self.dtype.enum])
         check_call(_LIB.set_tensor_buffer(self.tensor, np.ctypeslib.as_ctypes(value), value.size * value_bytes))
 
     def getData(self):