Fix a bug when reinit tensor, we didn't do so with stride.

liuliu · liuliu · commit 98eb262b1f22 · 2024-12-03T16:38:33.000-05:00
diff --git a/lib/nnc/ccv_nnc_symbolic_graph_compile.c b/lib/nnc/ccv_nnc_symbolic_graph_compile.c
@@ -4266,7 +4266,10 @@ int ccv_nnc_tensor_arena_reinit(ccv_nnc_tensor_arena_t* const tensor_arena, cons
 				const int alias_ref = tensor_arena->vt_alias_refs[i] - 1;
 				ccv_nnc_tensor_data(tensor->info, tensor_arena->vt_tensors[alias_ref]->data.u8, off + tensor_arena->vt_tensors[alias_ref]->dataof, &tensor->data, &tensor->dataof);
 				if (CCV_IS_TENSOR_VIEW(tensor))
+				{
 					((ccv_nnc_tensor_view_t*)tensor)->off = off;
+					memcpy(((ccv_nnc_tensor_view_t*)tensor)->stride, symbol_info->stride, sizeof(((ccv_nnc_tensor_view_t*)tensor)->stride));
+				}
 			}
 		}
 	// Should handle sub_tensor_arena, don't do that at the moment.

Original file line number	Diff line number	Diff line change
`@@ -4266,7 +4266,10 @@ int ccv_nnc_tensor_arena_reinit(ccv_nnc_tensor_arena_t* const tensor_arena, cons`
`4266`	`4266`	`const int alias_ref = tensor_arena->vt_alias_refs[i] - 1;`
`4267`	`4267`	`ccv_nnc_tensor_data(tensor->info, tensor_arena->vt_tensors[alias_ref]->data.u8, off + tensor_arena->vt_tensors[alias_ref]->dataof, &tensor->data, &tensor->dataof);`
`4268`	`4268`	`if (CCV_IS_TENSOR_VIEW(tensor))`
	`4269`	`+ {`
`4269`	`4270`	`((ccv_nnc_tensor_view_t*)tensor)->off = off;`
	`4271`	`+ memcpy(((ccv_nnc_tensor_view_t)tensor)->stride, symbol_info->stride, sizeof(((ccv_nnc_tensor_view_t)tensor)->stride));`
	`4272`	`+ }`
`4270`	`4273`	`}`
`4271`	`4274`	`}`
`4272`	`4275`	`// Should handle sub_tensor_arena, don't do that at the moment.`