Home
last modified time | relevance | path

Searched refs:cudaSetDevice (Results 1 – 13 of 13) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h140 cudaError_t err = cudaSetDevice(device_); in initializeDeviceProp()
150 cudaError_t err = cudaSetDevice(device_); in initializeDeviceProp()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_utils.cc175 [&] { XLA_CUDA_WARN_IF_ERROR(cudaSetDevice(initial_cuda_device)); }); in CreateNcclClique()
184 cudaSetDevice(local_participants[i].device_ordinal)); in CreateNcclClique()
Dnccl_utils.h43 #define cudaSetDevice hipSetDevice macro
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.cc48 #define cudaSetDevice hipSetDevice
408 CUDA_RETURN_IF_ERROR(cudaSetDevice(devices[i])); in GetCommunicator()
413 CUDA_RETURN_IF_ERROR(cudaSetDevice(saved_device)); in GetCommunicator()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc446 cudaSetDevice(cuda_device_id); in CreateTRTNode()
826 cudaSetDevice(old_cuda_device.value()); in ConvertAfterShapes()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1218 err = cudaSetDevice(platform_gpu_id.value()); in CreateDevices()
1271 err = cudaSetDevice(original_device); in CreateDevices()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc1062 auto err = cudaSetDevice(platform_gpu_id); in AllocateCalibrationResources()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_runtime_9_0.inc255 extern __host__ cudaError_t CUDARTAPI cudaSetDevice(int device) {
257 static auto func_ptr = LoadSymbol<FuncPtr>("cudaSetDevice");
Dcuda_runtime_10_0.inc256 extern __host__ cudaError_t CUDARTAPI cudaSetDevice(int device) {
258 static auto func_ptr = LoadSymbol<FuncPtr>("cudaSetDevice");
Dcuda_runtime_10_1.inc256 extern __host__ cudaError_t CUDARTAPI cudaSetDevice(int device) {
258 static auto func_ptr = LoadSymbol<FuncPtr>("cudaSetDevice");
Dcuda_runtime_10_2.inc265 extern __host__ cudaError_t CUDARTAPI cudaSetDevice(int device) {
267 static auto func_ptr = LoadSymbol<FuncPtr>("cudaSetDevice");
Dcuda_runtime_11_0.inc265 extern __host__ cudaError_t CUDARTAPI cudaSetDevice(int device) {
267 static auto func_ptr = LoadSymbol<FuncPtr>("cudaSetDevice");
Dcuda_runtime_11_2.inc288 extern __host__ cudaError_t CUDARTAPI cudaSetDevice(int device) {
290 static auto func_ptr = LoadSymbol<FuncPtr>("cudaSetDevice");