Home
last modified time | relevance | path

Searched refs:cuda_error (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/c10/cuda/
DCUDAException.cpp17 const auto cuda_error = static_cast<cudaError_t>(err); in c10_cuda_check_implementation() local
22 if (C10_LIKELY(cuda_error == cudaSuccess && !cuda_kernel_failure)) { in c10_cuda_check_implementation()
32 check_message.append(cudaGetErrorString(cuda_error)); in c10_cuda_check_implementation()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc876 cudaError_t cuda_error = cudaGetDevice(&cuda_device_id); in ConvertGraph() local
877 if (cuda_error != cudaSuccess) { in ConvertGraph()
879 << cudaGetErrorString(cuda_error); in ConvertGraph()