Home
last modified time | relevance | path

Searched refs:gpu_assert (Results 1 – 9 of 9) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceGpu.h80 gpu_assert(status == gpuSuccess); in initialize()
91 gpu_assert(status == gpuSuccess); in initialize()
147 gpu_assert(err == gpuSuccess); in stream_()
148 gpu_assert(device < num_devices); in stream_()
166 gpu_assert(err == gpuSuccess); in allocate()
169 gpu_assert(err == gpuSuccess); in allocate()
170 gpu_assert(result != NULL); in allocate()
176 gpu_assert(err == gpuSuccess); in deallocate()
177 gpu_assert(buffer != NULL); in deallocate()
179 gpu_assert(err == gpuSuccess); in deallocate()
[all …]
DTensorGpuHipCudaDefines.h88 #ifndef gpu_assert
92 #define gpu_assert(COND) macro
94 #define gpu_assert(COND) assert(COND) macro
DTensorReductionGpu.h63 gpu_assert(0 && "Wordsize not supported"); in atomicReduce()
66 gpu_assert(0 && "Shouldn't be called on unsupported device"); in atomicReduce()
117 gpu_assert(0 && "Shouldn't be called on unsupported device"); in atomicReduce()
208 gpu_assert(0 && "Shouldn't be called on unsupported device"); in FullReductionKernel()
378 gpu_assert(false && "Should only be called on doubles, floats and half floats"); in run()
411 gpu_assert(false && "Should not be called since there is no packet accessor");
464gpu_assert(HasOptimizedImplementation && "Should only be called on doubles, floats or half floats"…
555 gpu_assert(0 && "Shouldn't be called on unsupported device");
749gpu_assert(false && "Should only be called to reduce doubles, floats and half floats on a gpu devi…
796 gpu_assert(false && "Should not be called since there is no packet accessor");
[all …]
DTensorConvolutionSycl.h401 gpu_assert(static_cast<unsigned long>(local_memory_size) <= m_device.sharedMemPerBlock());
433 gpu_assert(static_cast<unsigned long>(local_memory_size) <= m_device.sharedMemPerBlock());
477 gpu_assert(static_cast<unsigned long>(local_memory_size) <= m_device.sharedMemPerBlock());
DTensorRandom.h23 gpu_assert(threadIdx.z == 0); in get_random_seed()
DTensorConvolution.h920 gpu_assert(shared_mem <= maxSharedMem);
977 gpu_assert(shared_mem <= maxSharedMem);
1057 gpu_assert(shared_mem <= maxSharedMem);
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.h25 #ifndef gpu_assert
26 #define gpu_assert(x) macro
/external/eigen/test/
Dgpu_common.h75 gpu_assert(false); in run_on_gpu()
82 gpu_assert(false); in run_on_gpu()
/external/eigen/unsupported/Eigen/CXX11/
DTensor65 #ifndef gpu_assert
66 #define gpu_assert(x)