Searched refs:cudaMalloc (Results 1 – 9 of 9) sorted by relevance
| /third_party/boost/libs/fiber/examples/cuda/ |
| D | multiple_streams.cu | 49 cudaMalloc( & dev_a0, size * sizeof( int) ); in main()
50 cudaMalloc( & dev_b0, size * sizeof( int) ); in main()
51 cudaMalloc( & dev_c0, size * sizeof( int) ); in main()
52 cudaMalloc( & dev_a1, size * sizeof( int) ); in main()
53 cudaMalloc( & dev_b1, size * sizeof( int) ); in main()
54 cudaMalloc( & dev_c1, size * sizeof( int) ); in main()
|
| D | single_stream.cu | 46 cudaMalloc( & dev_a, size * sizeof( int) ); in main()
47 cudaMalloc( & dev_b, size * sizeof( int) ); in main()
48 cudaMalloc( & dev_c, size * sizeof( int) ); in main()
|
| /third_party/boost/libs/fiber/doc/ |
| D | cuda.qbk | 47 cudaMalloc( & dev_a, size * sizeof( int) );
48 cudaMalloc( & dev_b, size * sizeof( int) );
49 cudaMalloc( & dev_c, size * sizeof( int) );
|
| /third_party/mindspore/tests/ut/cpp/stub/runtime/ |
| D | cuda_runtime_api.cc | 18 cudaError_t cudaMalloc(void **devPtr, size_t size) { return cudaSuccess; } in cudaMalloc() function
|
| D | cuda_runtime_api.h | 37 cudaError_t cudaMalloc(void **devPtr, size_t size);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | ctcloss_gpu_kernel.h | 258 … cudaMalloc(reinterpret_cast<void **>(log_alpha_b), sizeof(T) * log_prob_size), in MemManageForCus()
261 …kernel_node_, cudaMalloc(reinterpret_cast<void **>(log_beta_b), sizeof(T) * log_prob_size), "cudaM… in MemManageForCus()
269 …cudaMalloc(reinterpret_cast<void **>(label_value_with_blank), sizeof(int) * (2 * total_labels_size… in MemManageForCus()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | cuda_driver.cc | 27 auto ret = cudaMalloc(reinterpret_cast<void **>(addr), size); in AllocDeviceMem()
32 ret = cudaMalloc(reinterpret_cast<void **>(addr), size); in AllocDeviceMem()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_allocator.cc | 37 auto cuda_ret = cudaMalloc(&device_ptr, size); in MallocDeviceMem()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/ |
| D | common_sponge.cuh | 244 cudaError_t cuda_error = cudaMalloc(&address[0], size); in Cuda_Malloc_Safely()
|