Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/
DDispatchStub.cpp153 … cuda_dispatch_ptr != nullptr ? DispatchResult(cuda_dispatch_ptr) : ErrorType::MissingDeviceKernel; in try_get_call_ptr()
156 …rn hip_dispatch_ptr != nullptr ? DispatchResult(hip_dispatch_ptr) : ErrorType::MissingDeviceKernel; in try_get_call_ptr()
160 …rn mps_dispatch_ptr != nullptr ? DispatchResult(mps_dispatch_ptr) : ErrorType::MissingDeviceKernel; in try_get_call_ptr()
163 … mtia_dispatch_ptr != nullptr ? DispatchResult(mtia_dispatch_ptr) : ErrorType::MissingDeviceKernel; in try_get_call_ptr()
167 …rn xpu_dispatch_ptr != nullptr ? DispatchResult(xpu_dispatch_ptr) : ErrorType::MissingDeviceKernel; in try_get_call_ptr()
171 …ispatch_ptr != nullptr ? DispatchResult(privateuse1_dispatch_ptr) : ErrorType::MissingDeviceKernel; in try_get_call_ptr()
219 case ErrorType::MissingDeviceKernel: in get_call_ptr()
257 return AVX2 != nullptr ? DispatchResult(AVX2) : ErrorType::MissingDeviceKernel; in try_choose_cpu_impl()
265 return AVX2 != nullptr ? DispatchResult(AVX2) : ErrorType::MissingDeviceKernel; in try_choose_cpu_impl()
270 return VSX != nullptr ? DispatchResult(VSX) : ErrorType::MissingDeviceKernel; in try_choose_cpu_impl()
[all …]
DDispatchStub.h76 MissingDeviceKernel, enumerator