| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_platform_id.cc | 16 #include "tensorflow/stream_executor/cuda/cuda_platform_id.h"
19 namespace cuda { namespace in namespace:stream_executor
23 } // namespace cuda
|
| cuda_activation.h | 16 // This file contains APIs that assume a StreamExecutor is backed by CUDA.
17 // It reaches into the CUDA implementation to activate an underlying CUDA
20 // Having this file separate from cuda/cuda_gpu_executor.h means that dependent
21 // code does not also have to depend on cuda.h.
32 namespace cuda { namespace in namespace:stream_executor
36 } // namespace cuda
|
| cuda_event.h | 22 namespace cuda { namespace in namespace:stream_executor
26 } // namespace cuda
|
| cuda_gpu_executor.h | 16 // The CUDA implementation of the StreamExecutorInterface functionality.
17 // CUDA inclusions are ideally confined to this implementation file.
19 // The notions from the StreamExecutor basically correspond to the CUDA streams
28 namespace cuda { namespace in namespace:stream_executor
32 } // namespace cuda
|
| cuda_kernel.h | 16 // The CUDA implementation of the StreamExecutorInterface functionality.
17 // CUDA inclusions are ideally confined to this implementation file.
19 // The notions from the StreamExecutor basically correspond to the CUDA streams
28 namespace cuda { namespace in namespace:stream_executor
32 } // namespace cuda
|
| cuda_platform_id.h | 22 namespace cuda { namespace in namespace:stream_executor
24 // Opaque and unique identifier for the cuda platform.
31 } // namespace cuda
|
| cuda_rng.h | 23 namespace cuda { namespace in namespace:stream_executor
27 } // namespace cuda
|
| cuda_timer.h | 16 // Defines the GpuTimer type - the CUDA-specific implementation of the generic
25 namespace cuda { namespace in namespace:stream_executor
29 } // namespace cuda
|
| cuda_stream.h | 16 // Defines the GpuStream type - the CUDA-specific implementation of the generic
25 namespace cuda { namespace in namespace:stream_executor
33 } // namespace cuda
|
| cuda_diagnostics.h | 22 namespace cuda { namespace in namespace:stream_executor
38 } // namespace cuda
|
| cuda_driver.h | 16 // CUDA userspace driver library wrapper functionality.
25 // CUDAContext wraps a cuda CUcontext handle, and includes a unique id. The
47 namespace cuda { namespace in namespace:stream_executor
57 // Returns the current context set in CUDA. This is done by calling the cuda
61 } // namespace cuda
|
| cuda_diagnostics.cc | 16 #include "tensorflow/stream_executor/cuda/cuda_diagnostics.h"
53 namespace cuda { namespace in namespace:stream_executor
108 } // namespace cuda
115 static const CFStringRef kDriverKextIdentifier = CFSTR("com.nvidia.CUDA");
166 LOG(INFO) << "retrieving CUDA diagnostic information for host: "
198 << cuda::DriverVersionStatusToString(dso_version);
202 << cuda::DriverVersionStatusToString(kernel_version);
221 // OSX CUDA libraries have names like: libcuda_310.41.15_mercury.dylib
240 result = cuda::StringToDriverVersion(version);
268 *result = cuda::StringToDriverVersion(stripped_dso_version)
[all...] |
| cuda_platform.h | 35 namespace cuda { namespace in namespace:stream_executor
36 // Opaque and unique identifier for the CUDA platform plugin.
40 } // namespace cuda
43 // Cuda-specific platform plugin, registered as a singleton value via module
108 namespace cuda { namespace in namespace:stream_executor
112 } // namespace cuda
|
| cuda_rng.cc | 16 #include "tensorflow/stream_executor/cuda/cuda_rng.h"
18 #include "tensorflow/stream_executor/cuda/cuda_activation.h"
19 #include "tensorflow/stream_executor/cuda/cuda_gpu_executor.h"
20 #include "tensorflow/stream_executor/cuda/cuda_helpers.h"
21 #include "tensorflow/stream_executor/cuda/cuda_platform_id.h"
22 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
30 #include "cuda/include/curand.h"
68 cuda::ScopedActivateExecutorContext sac(parent_);
77 cuda::ScopedActivateExecutorContext sac(parent_);
89 cuda::ScopedActivateExecutorContext sac(parent_)
[all...] |
| /external/skia/src/compute/common/cuda/ |
| assert_cuda.h | 32 #define cuda(...) assert_cuda((cuda##__VA_ARGS__), __FILE__, __LINE__, true); macro
|
| /external/skqp/src/compute/common/cuda/ |
| assert_cuda.h | 32 #define cuda(...) assert_cuda((cuda##__VA_ARGS__), __FILE__, __LINE__, true); macro
|
| /external/tensorflow/tensorflow/tools/dockerfiles/tests/ |
| build-gpu.sh | 25 ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1
27 LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs:${LD_LIBRARY_PATH} \
29 bazel build -c opt --copt=-mavx --config=cuda \
32 rm /usr/local/cuda/lib64/stubs/libcuda.so.1 && \
|
| /external/skia/src/compute/hs/cuda/bench/ |
| main.c | 30 #include "common/cuda/assert_cuda.h"
37 #include "hs/cuda/sm_35/u32/hs_cuda.h"
38 #include "hs/cuda/sm_35/u64/hs_cuda.h"
293 cuda(Malloc(&random_d,size_hi_in));
294 cuda(Malloc(&vin_d, size_hi_in));
295 cuda(Malloc(&vout_d, size_hi_out));
306 cuda(Memcpy(random_d,random_h,size_hi_in,cudaMemcpyHostToDevice));
342 cuda(EventCreate(&start));
343 cuda(EventCreate(&end));
352 cuda(Memcpy(vin_d,random_d,count*key_size,cudaMemcpyDeviceToDevice))
[all...] |
| /external/skqp/src/compute/hs/cuda/bench/ |
| main.c | 30 #include "common/cuda/assert_cuda.h"
37 #include "hs/cuda/sm_35/u32/hs_cuda.h"
38 #include "hs/cuda/sm_35/u64/hs_cuda.h"
293 cuda(Malloc(&random_d,size_hi_in));
294 cuda(Malloc(&vin_d, size_hi_in));
295 cuda(Malloc(&vout_d, size_hi_out));
306 cuda(Memcpy(random_d,random_h,size_hi_in,cudaMemcpyHostToDevice));
342 cuda(EventCreate(&start));
343 cuda(EventCreate(&end));
352 cuda(Memcpy(vin_d,random_d,count*key_size,cudaMemcpyDeviceToDevice))
[all...] |
| /external/tensorflow/tensorflow/contrib/makefile/ |
| build_all_android.sh | 70 if [ ! -d ${JETPACK}/cuda ]; then
71 ln -s $(ls -d ${JETPACK}/cuda-*/|sort -r|head -n1) ${JETPACK}/cuda
73 if [ ! -d ${JETPACK}/cuda ]; then
74 ln -s $(ls -d ${JETPACK}/cuda-*/|sort -r|head -n1) ${JETPACK}/cuda
|
| /external/tensorflow/tensorflow/tools/ci_build/windows/gpu/bazel/ |
| run_cc_test_windows.sh | 59 # bazel test --config=win-cuda -k $slow_compiling_test --test_output=errors
60 bazel test -c opt --config=win-cuda -k $passing_tests --test_output=errors
|
| /external/skia/src/compute/hs/cuda/sm_35/u32/ |
| hs_cuda.h | 15 #include <cuda.h>
|
| /external/skia/src/compute/hs/cuda/sm_35/u64/ |
| hs_cuda.h | 15 #include <cuda.h>
|
| /external/skqp/src/compute/hs/cuda/sm_35/u32/ |
| hs_cuda.h | 15 #include <cuda.h>
|
| /external/skqp/src/compute/hs/cuda/sm_35/u64/ |
| hs_cuda.h | 15 #include <cuda.h>
|
