Home
last modified time | relevance | path

Searched refs:__CUDA_ARCH__ (Results 1 – 25 of 37) sorted by relevance

12

/external/eigen/Eigen/src/Core/arch/CUDA/
DHalf.h143 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530
284 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300 in float_to_half_rtne()
339 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300 in half_to_float()
373 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530
392 …defined __CUDACC_VER__ && __CUDACC_VER__ >= 80000 && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530 in log()
430 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530 in half()
439 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530 in half()
560 #if defined __CUDACC_VER__ && __CUDACC_VER__ >= 80000 && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >=…
594 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
601 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 350
[all …]
DTypeCasting.h22 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
40 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
58 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
72 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
DPacketMath.h199 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 350
207 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 350
216 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 350
224 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 350
DPacketMathHalf.h18 #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDACC__) && defined(__CUDA_ARCH__) && __CUDA_ARCH__
71 #if __CUDA_ARCH__ >= 350
80 #if __CUDA_ARCH__ >= 350
118 #if __CUDA_ARCH__ >= 530
127 #if __CUDA_ARCH__ >= 530
141 #if __CUDA_ARCH__ >= 530
155 #if __CUDA_ARCH__ >= 530
167 #if __CUDA_ARCH__ >= 530
181 #if __CUDA_ARCH__ >= 530
227 #if __CUDA_ARCH__ >= 530
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceDefault.h38 #ifndef __CUDA_ARCH__ in numThreads()
48 #ifndef __CUDA_ARCH__ in firstLevelCacheSize()
58 #ifndef __CUDA_ARCH__ in lastLevelCacheSize()
68 #ifndef __CUDA_ARCH__ in majorDeviceVersion()
74 return __CUDA_ARCH__ / 100; in majorDeviceVersion()
DTensorIntDiv.h38 #ifdef __CUDA_ARCH__ in count_leading_zeros()
54 #ifdef __CUDA_ARCH__ in count_leading_zeros()
89 #if defined(__CUDA_ARCH__) in muluh()
98 #if defined(__CUDA_ARCH__) in muluh()
119 #if defined(__SIZEOF_INT128__) && !defined(__CUDA_ARCH__)
198 #ifdef __CUDA_ARCH__
DTensorDeviceCuda.h214 #ifndef __CUDA_ARCH__ in initializeDeviceProp()
239 #ifndef __CUDA_ARCH__ in initializeDeviceProp()
265 #if defined(__CUDACC__) && !defined(__CUDA_ARCH__) in initializeDeviceProp()
325 #ifndef __CUDA_ARCH__ in initializeDeviceProp()
DTensorReductionCuda.h26 #if __CUDA_ARCH__ >= 300 in atomicReduce()
105 #if __CUDA_ARCH__ >= 300 in atomicReduce()
127 #if __CUDA_ARCH__ >= 300 in FullReductionKernel()
376 #if __CUDA_ARCH__ >= 300
/external/clang/test/Preprocessor/
Dcuda-preprocess.cu7 #ifndef __CUDA_ARCH__
11 clang_unittest_cuda_arch __CUDA_ARCH__
/external/clang/test/SemaCUDA/
Dfunction-overload.cu24 #ifdef __CUDA_ARCH__
185 #if defined(__CUDA_ARCH__) in hostdevicef()
190 #if !defined(__CUDA_ARCH__) in hostdevicef()
320 #ifdef __CUDA_ARCH__ in test_host_device_calls_template()
Dqualifiers.cu20 #if defined(__CUDA_ARCH__)
Dasm-constraints-mixed.cu31 #if defined(__CUDA_ARCH__)
Dvararg.cu27 #if !defined(__CUDA_ARCH__) in baz()
Dhost-device-constexpr.cu64 #ifdef __CUDA_ARCH__ in HostDeviceFn()
/external/clang/lib/Headers/
D__clang_cuda_intrinsics.h34 #if !defined(__CUDA_ARCH__) || __CUDA_ARCH__ >= 300
101 #if !defined(__CUDA_ARCH__) || __CUDA_ARCH__ >= 320
D__clang_cuda_runtime_wrapper.h71 #ifndef __CUDA_ARCH__
72 #define __CUDA_ARCH__ 350 macro
/external/tensorflow/tensorflow/core/util/
Dcuda_device_functions.h349 #if __CUDA_ARCH__ >= 350 in CudaLdg()
362 #if __CUDA_ARCH__ >= 350 in CudaLdg()
372 #if __CUDA_ARCH__ >= 350 in CudaLdg()
497 #if __CUDA_ARCH__ < 600
582 #if __CUDA_ARCH__ < 320
612 #if __CUDA_ARCH__ < 320
/external/eigen/test/
Dcuda_common.h70 #ifndef __CUDA_ARCH__ in run_and_compare_to_cuda()
76 #ifndef __CUDA_ARCH__ in run_and_compare_to_cuda()
Dcuda_basic.cu149 #ifndef __CUDA_ARCH__ in test_cuda_basic()
/external/clang/test/CodeGenCUDA/
Dfunction-overload.cu29 #if defined(__CUDA_ARCH__)
/external/eigen/Eigen/src/Core/util/
DMeta.h14 #if defined(__CUDA_ARCH__)
167 #if defined(__CUDA_ARCH__)
468 #if defined(__CUDA_ARCH__)
474 #if defined(__CUDA_ARCH__)
/external/eigen/Eigen/src/Core/
DMathFunctions.h100 #ifdef __CUDA_ARCH__
148 #ifdef __CUDA_ARCH__
733 #ifdef __CUDA_ARCH__
748 #ifdef __CUDA_ARCH__
763 #ifdef __CUDA_ARCH__
829 #ifndef __CUDA_ARCH__
/external/llvm/docs/
DCompileCudaWithLLVM.rst133 subtool of NVCC's -- it defines the ``__CUDA__`` macro. ``__CUDA_ARCH__`` is
140 #if defined(__clang__) && defined(__CUDA__) && !defined(__CUDA_ARCH__)
144 #if defined(__clang__) && defined(__CUDA__) && defined(__CUDA_ARCH__)
/external/tensorflow/tensorflow/core/kernels/
Drelu_op_gpu.cu.cc51 #if __CUDA_ARCH__ >= 530 in ReluGradHalfKernel()
/external/tensorflow/tensorflow/core/lib/random/
Dphilox_random.h216 #ifndef __CUDA_ARCH__ in MultiplyHighLow()

12