Commit e1769a75 authored by fenglei.tian's avatar fenglei.tian

fix void pointer cast bug

parent 40e64c21
...@@ -511,7 +511,7 @@ void runtime::gpu::GPU_ExternalFunction::compile() ...@@ -511,7 +511,7 @@ void runtime::gpu::GPU_ExternalFunction::compile()
{ {
stringstream ss; stringstream ss;
ss << "((" << tensor->get_element_type().c_type_string() ss << "((" << tensor->get_element_type().c_type_string()
<< "*)(pool_base_ptr + " << tensor->get_pool_offset() << "))"; << "*)((char *)pool_base_ptr + " << tensor->get_pool_offset() << "))";
m_variable_name_map[tensor->get_name()] = ss.str(); m_variable_name_map[tensor->get_name()] = ss.str();
} }
} }
......
...@@ -24,7 +24,7 @@ namespace ngraph ...@@ -24,7 +24,7 @@ namespace ngraph
{ {
void print_gpu_f32_tensor(void* p, size_t element_count, size_t element_size); void print_gpu_f32_tensor(void* p, size_t element_count, size_t element_size);
void check_cuda_errors(CUresult err); void check_cuda_errors(CUresult err);
void** create_gpu_buffer(size_t buffer_size); void* create_gpu_buffer(size_t buffer_size);
void cuda_memcpyDtD(void* d, void* s, size_t element_count, size_t element_size); void cuda_memcpyDtD(void* d, void* s, size_t element_count, size_t element_size);
void cuda_memcpyHtD(void* d, void* s, size_t buffer_size); void cuda_memcpyHtD(void* d, void* s, size_t buffer_size);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment