Home
last modified time | relevance | path

Searched refs:CUDA_KERNEL_ASSERT (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dcuda_common.h23 #define CUDA_KERNEL_ASSERT(cond) \ macro
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dgather_grad.cu37 CUDA_KERNEL_ASSERT(j >= 0); in GatherGradKernel()
39 CUDA_KERNEL_ASSERT(j_read < dim_at_axis_output); in GatherGradKernel()
Dgather.cu35 CUDA_KERNEL_ASSERT(j >= 0); in GatherKernel()
37 CUDA_KERNEL_ASSERT(j_read < dim_at_axis_input); in GatherKernel()