Skip to content

Commit 62ba7f7

Browse files
committed
fix build
1 parent 289c329 commit 62ba7f7

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

model.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2222,14 +2222,15 @@ bool ModelLoader::load_tensors(on_new_tensor_cb_t on_new_tensor_cb, int n_thread
22222222

22232223
if (tensor_storage.type == dst_tensor->type) {
22242224
// copy to device memory
2225-
curr_time_ms = ggml_time_ms();
2226-
convert_time_ms += curr_time_ms - prev_time_ms;
2227-
prev_time_ms = curr_time_ms;
2225+
t1 = ggml_time_ms();
2226+
convert_time_ms.fetch_add(t1 - t0);
2227+
t0 = ggml_time_ms();
22282228
ggml_backend_tensor_set(dst_tensor, read_buffer.data(), 0, ggml_nbytes(dst_tensor));
22292229
t1 = ggml_time_ms();
22302230
copy_to_backend_time_ms.fetch_add(t1 - t0);
22312231
} else {
22322232
// convert first, then copy to device memory
2233+
22332234
convert_buffer.resize(ggml_nbytes(dst_tensor));
22342235
convert_tensor((void*)read_buffer.data(), tensor_storage.type, (void*)convert_buffer.data(), dst_tensor->type, (int)tensor_storage.nelements() / (int)tensor_storage.ne[0], (int)tensor_storage.ne[0]);
22352236
t1 = ggml_time_ms();

0 commit comments

Comments
 (0)