2
2
#include " ggml.h"
3
3
#include " ggml-backend-impl.h"
4
4
5
+ #include < cinttypes>
5
6
#include < string>
6
7
#include < vector>
7
8
#include < memory>
@@ -729,7 +730,7 @@ static void rpc_alloc_buffer(ggml_backend_t backend, const std::vector<uint8_t>
729
730
ggml_backend_buffer_t buffer = ggml_backend_buft_alloc_buffer (buft, size);
730
731
uint64_t remote_ptr = reinterpret_cast <uint64_t >(buffer);
731
732
uint64_t remote_size = buffer->size ;
732
- GGML_PRINT_DEBUG (" [%s] size: %lu -> remote_ptr: %lx , remote_size: %lu \n " , __func__, size, remote_ptr, remote_size);
733
+ GGML_PRINT_DEBUG (" [%s] size: %" PRIu64 " -> remote_ptr: %" PRIx64 " , remote_size: %" PRIu64 " \n " , __func__, size, remote_ptr, remote_size);
733
734
// output serialization format: | remote_ptr (8 bytes) | remote_size (8 bytes) |
734
735
output.resize (2 *sizeof (uint64_t ), 0 );
735
736
memcpy (output.data (), &remote_ptr, sizeof (remote_ptr));
@@ -758,7 +759,7 @@ static void rpc_buffer_get_base(const std::vector<uint8_t> & input, std::vector<
758
759
// input serialization format: | remote_ptr (8 bytes) |
759
760
uint64_t remote_ptr;
760
761
memcpy (&remote_ptr, input.data (), sizeof (remote_ptr));
761
- GGML_PRINT_DEBUG (" [%s] remote_ptr: %lx \n " , __func__, remote_ptr);
762
+ GGML_PRINT_DEBUG (" [%s] remote_ptr: %" PRIx64 " \n " , __func__, remote_ptr);
762
763
ggml_backend_buffer_t buffer = reinterpret_cast <ggml_backend_buffer_t >(remote_ptr);
763
764
void * base = ggml_backend_buffer_get_base (buffer);
764
765
// output serialization format: | base_ptr (8 bytes) |
@@ -771,7 +772,7 @@ static void rpc_free_buffer(const std::vector<uint8_t> & input) {
771
772
// input serialization format: | remote_ptr (8 bytes) |
772
773
uint64_t remote_ptr;
773
774
memcpy (&remote_ptr, input.data (), sizeof (remote_ptr));
774
- GGML_PRINT_DEBUG (" [%s] remote_ptr: %lx \n " , __func__, remote_ptr);
775
+ GGML_PRINT_DEBUG (" [%s] remote_ptr: %" PRIx64 " \n " , __func__, remote_ptr);
775
776
ggml_backend_buffer_t buffer = reinterpret_cast <ggml_backend_buffer_t >(remote_ptr);
776
777
ggml_backend_buffer_free (buffer);
777
778
}
@@ -782,7 +783,7 @@ static void rpc_buffer_clear(const std::vector<uint8_t> & input) {
782
783
memcpy (&remote_ptr, input.data (), sizeof (remote_ptr));
783
784
uint8_t value;
784
785
memcpy (&value, input.data () + sizeof (uint64_t ), sizeof (value));
785
- GGML_PRINT_DEBUG (" [%s] remote_ptr: %lx , value: %u\n " , __func__, remote_ptr, value);
786
+ GGML_PRINT_DEBUG (" [%s] remote_ptr: %" PRIx64 " , value: %u\n " , __func__, remote_ptr, value);
786
787
ggml_backend_buffer_t buffer = reinterpret_cast <ggml_backend_buffer_t >(remote_ptr);
787
788
ggml_backend_buffer_clear (buffer, value);
788
789
}
@@ -801,7 +802,7 @@ static void rpc_set_tensor(const std::vector<uint8_t> & input) {
801
802
};
802
803
struct ggml_context * ctx = ggml_init (params);
803
804
ggml_tensor * tensor = deserialize_tensor (ctx, in_tensor);
804
- GGML_PRINT_DEBUG (" [%s] buffer: %p, data: %p, offset: %lu , size: %lu \n " , __func__, (void *)tensor->buffer , tensor->data , offset, size);
805
+ GGML_PRINT_DEBUG (" [%s] buffer: %p, data: %p, offset: %" PRIu64 " , size: %zu \n " , __func__, (void *)tensor->buffer , tensor->data , offset, size);
805
806
const void * data = input.data () + sizeof (rpc_tensor) + sizeof (offset);
806
807
ggml_backend_tensor_set (tensor, data, offset, size);
807
808
ggml_free (ctx);
@@ -822,7 +823,7 @@ static void rpc_get_tensor(const std::vector<uint8_t> & input, std::vector<uint8
822
823
};
823
824
struct ggml_context * ctx = ggml_init (params);
824
825
ggml_tensor * tensor = deserialize_tensor (ctx, in_tensor);
825
- GGML_PRINT_DEBUG (" [%s] buffer: %p, data: %p, offset: %lu , size: %lu \n " , __func__, (void *)tensor->buffer , tensor->data , offset, size);
826
+ GGML_PRINT_DEBUG (" [%s] buffer: %p, data: %p, offset: %" PRIu64 " , size: %" PRIu64 " \n " , __func__, (void *)tensor->buffer , tensor->data , offset, size);
826
827
// output serialization format: | data (size bytes) |
827
828
output.resize (size, 0 );
828
829
ggml_backend_tensor_get (tensor, output.data (), offset, size);
0 commit comments