| /external/tensorflow/tensorflow/core/platform/default/build_config/ |
| D | BUILD | 38 "//tensorflow/stream_executor/cuda:cuda_activation",
47 "//tensorflow/stream_executor/cuda:cuda_platform_id",
52 "@local_config_cuda//cuda:darwin": ["IOKit"],
55 "//tensorflow:using_cuda_clang": ["//tensorflow/stream_executor/cuda:all_runtime"],
56 "//tensorflow:using_cuda_nvcc": ["//tensorflow/stream_executor/cuda:all_runtime"],
68 ":cuda",
70 ["//tensorflow/stream_executor/cuda:all_runtime"],
72 "@local_config_cuda//cuda:darwin": ["IOKit"],
97 "//tensorflow/stream_executor/cuda:cuda_platform_id",
105 # Dummy stream executor cuda plugins.
[all …]
|
| /external/skqp/src/compute/hs/cuda/bench/ |
| D | main.c | 293 cuda(Malloc(&random_d,size_hi_in)); in hs_bench()
294 cuda(Malloc(&vin_d, size_hi_in)); in hs_bench()
295 cuda(Malloc(&vout_d, size_hi_out)); in hs_bench()
306 cuda(Memcpy(random_d,random_h,size_hi_in,cudaMemcpyHostToDevice)); in hs_bench()
342 cuda(EventCreate(&start)); in hs_bench()
343 cuda(EventCreate(&end)); in hs_bench()
352 cuda(Memcpy(vin_d,random_d,count*key_size,cudaMemcpyDeviceToDevice)); in hs_bench()
370 cuda(EventRecord(start,stream0)); in hs_bench()
371 cuda(StreamWaitEvent(stream1,start,0)); in hs_bench()
372 cuda(StreamWaitEvent(stream2,start,0)); in hs_bench()
[all …]
|
| /external/skia/src/compute/hs/cuda/bench/ |
| D | main.c | 293 cuda(Malloc(&random_d,size_hi_in)); in hs_bench()
294 cuda(Malloc(&vin_d, size_hi_in)); in hs_bench()
295 cuda(Malloc(&vout_d, size_hi_out)); in hs_bench()
306 cuda(Memcpy(random_d,random_h,size_hi_in,cudaMemcpyHostToDevice)); in hs_bench()
342 cuda(EventCreate(&start)); in hs_bench()
343 cuda(EventCreate(&end)); in hs_bench()
352 cuda(Memcpy(vin_d,random_d,count*key_size,cudaMemcpyDeviceToDevice)); in hs_bench()
370 cuda(EventRecord(start,stream0)); in hs_bench()
371 cuda(StreamWaitEvent(stream1,start,0)); in hs_bench()
372 cuda(StreamWaitEvent(stream2,start,0)); in hs_bench()
[all …]
|
| /external/tensorflow/tensorflow/tools/dockerfiles/partials/ubuntu/ |
| D | devel-nvidia.partial.Dockerfile | 3 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
16 cuda-command-line-tools-${CUDA/./-} \
17 cuda-cublas-dev-${CUDA/./-} \
18 cuda-cudart-dev-${CUDA/./-} \
19 cuda-cufft-dev-${CUDA/./-} \
20 cuda-curand-dev-${CUDA/./-} \
21 cuda-cusolver-dev-${CUDA/./-} \
22 cuda-cusparse-dev-${CUDA/./-} \
23 libcudnn7=${CUDNN}+cuda${CUDA} \
24 libcudnn7-dev=${CUDNN}+cuda${CUDA} \
[all …]
|
| D | nvidia.partial.Dockerfile | 3 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
15 cuda-command-line-tools-${CUDA/./-} \
16 cuda-cublas-${CUDA/./-} \
17 cuda-cufft-${CUDA/./-} \
18 cuda-curand-${CUDA/./-} \
19 cuda-cusolver-${CUDA/./-} \
20 cuda-cusparse-${CUDA/./-} \
22 libcudnn7=${CUDNN}+cuda${CUDA} \
31 apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
33 && apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda${CUDA} \
[all …]
|
| /external/tensorflow/tensorflow/tools/ci_build/ |
| D | Dockerfile.rbe.cuda9.0-cudnn7-ubuntu14.04 | 16 …apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_6…
19 …ttps://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.li…
24 ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
27 ENV NVIDIA_REQUIRE_CUDA "cuda>=9.0"
31 # TODO(b/110903506): /usr/loca/cuda/lib64/stubs should not be needed in
34 # -Wl,-rpath-link=/usr/local/cuda/lib64/stubs to all binaries transitively
35 # depending on libcuda. Optimally, builds targeting cuda would do that
37 ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64/stubs
40 LABEL com.nvidia.cuda.version="${CUDA_VERSION}"
44 cuda-cudart-$CUDA_PKG_VERSION \
[all …]
|
| D | Dockerfile.rbe.cuda10.0-cudnn7-ubuntu14.04 | 13 …apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_6…
16 …ttps://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.li…
24 ENV NVIDIA_REQUIRE_CUDA "cuda>=10.0,driver>=410"
26 ENV PATH /usr/local/cuda/bin:${PATH}
28 # TODO(b/110903506): /usr/loca/cuda/lib64/stubs should not be needed in
31 # -Wl,-rpath-link=/usr/local/cuda/lib64/stubs to all binaries transitively
32 # depending on libcuda. Optimally, builds targeting cuda would do that
34 ENV LD_LIBRARY_PATH /usr/local/cuda/lib64/stubs
39 cuda-command-line-tools-$CUDA_PKG_VERSION \
40 cuda-compat-10-0=410.48-1 \
[all …]
|
| D | Dockerfile.gpu | 1 FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
6 # /usr/local/cuda
7 RUN cp -P /usr/include/cudnn.h /usr/local/cuda/include
8 RUN cp -P /usr/lib/x86_64-linux-gnu/libcudnn* /usr/local/cuda/lib64
30 ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH
33 RUN mkdir /usr/local/cuda/lib && \
34 ln -s /usr/lib/x86_64-linux-gnu/libnccl.so.2 /usr/local/cuda/lib/libnccl.so.2 && \
35 ln -s /usr/include/nccl.h /usr/local/cuda/include/nccl.h
|
| D | Dockerfile.gpu.ppc64le | 1 FROM nvidia/cuda-ppc64le:9.2-cudnn7-devel-ubuntu16.04
6 # /usr/local/cuda
7 RUN cp -P /usr/include/cudnn.h /usr/local/cuda/include
8 RUN cp -P /usr/lib/powerpc64le-linux-gnu/libcudnn* /usr/local/cuda/lib64
24 ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH
30 ENV CUDA_TOOLKIT_PATH /usr/local/cuda-9.2
|
| /external/tensorflow/tensorflow/tools/docker/ |
| D | Dockerfile.devel-gpu | 1 FROM nvidia/cuda:9.0-base-ubuntu16.04
7 cuda-command-line-tools-9-0 \
8 cuda-cublas-dev-9-0 \
9 cuda-cudart-dev-9-0 \
10 cuda-cufft-dev-9-0 \
11 cuda-curand-dev-9-0 \
12 cuda-cusolver-dev-9-0 \
13 cuda-cusparse-dev-9-0 \
33 …find /usr/local/cuda-9.0/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -del…
100 ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH
[all …]
|
| D | Dockerfile.gpu | 1 FROM nvidia/cuda:9.0-base-ubuntu16.04
8 cuda-command-line-tools-9-0 \
9 cuda-cublas-9-0 \
10 cuda-cufft-9-0 \
11 cuda-curand-9-0 \
12 cuda-cusolver-9-0 \
13 cuda-cusparse-9-0 \
80 ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH
|
| /external/tensorflow/tensorflow/tools/dockerfiles/dockerfiles/ppc64le/ |
| D | devel-gpu-ppc64le.Dockerfile | 26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
39 cuda-command-line-tools-${CUDA/./-} \
40 cuda-cublas-dev-${CUDA/./-} \
41 cuda-cudart-dev-${CUDA/./-} \
42 cuda-cufft-dev-${CUDA/./-} \
43 cuda-curand-dev-${CUDA/./-} \
44 cuda-cusolver-dev-${CUDA/./-} \
45 cuda-cusparse-dev-${CUDA/./-} \
46 libcudnn7=${CUDNN}+cuda${CUDA} \
47 libcudnn7-dev=${CUDNN}+cuda${CUDA} \
[all …]
|
| D | devel-gpu-ppc64le-jupyter.Dockerfile | 26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
39 cuda-command-line-tools-${CUDA/./-} \
40 cuda-cublas-dev-${CUDA/./-} \
41 cuda-cudart-dev-${CUDA/./-} \
42 cuda-cufft-dev-${CUDA/./-} \
43 cuda-curand-dev-${CUDA/./-} \
44 cuda-cusolver-dev-${CUDA/./-} \
45 cuda-cusparse-dev-${CUDA/./-} \
46 libcudnn7=${CUDNN}+cuda${CUDA} \
47 libcudnn7-dev=${CUDNN}+cuda${CUDA} \
[all …]
|
| D | gpu-ppc64le.Dockerfile | 26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
38 cuda-command-line-tools-${CUDA/./-} \
39 cuda-cublas-${CUDA/./-} \
40 cuda-cufft-${CUDA/./-} \
41 cuda-curand-${CUDA/./-} \
42 cuda-cusolver-${CUDA/./-} \
43 cuda-cusparse-${CUDA/./-} \
45 libcudnn7=${CUDNN}+cuda${CUDA} \
54 apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
56 && apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda${CUDA} \
[all …]
|
| D | gpu-ppc64le-jupyter.Dockerfile | 26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
38 cuda-command-line-tools-${CUDA/./-} \
39 cuda-cublas-${CUDA/./-} \
40 cuda-cufft-${CUDA/./-} \
41 cuda-curand-${CUDA/./-} \
42 cuda-cusolver-${CUDA/./-} \
43 cuda-cusparse-${CUDA/./-} \
45 libcudnn7=${CUDNN}+cuda${CUDA} \
54 apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
56 && apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda${CUDA} \
[all …]
|
| /external/tensorflow/tensorflow/tools/dockerfiles/dockerfiles/ |
| D | devel-gpu.Dockerfile | 26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
39 cuda-command-line-tools-${CUDA/./-} \
40 cuda-cublas-dev-${CUDA/./-} \
41 cuda-cudart-dev-${CUDA/./-} \
42 cuda-cufft-dev-${CUDA/./-} \
43 cuda-curand-dev-${CUDA/./-} \
44 cuda-cusolver-dev-${CUDA/./-} \
45 cuda-cusparse-dev-${CUDA/./-} \
46 libcudnn7=${CUDNN}+cuda${CUDA} \
47 libcudnn7-dev=${CUDNN}+cuda${CUDA} \
[all …]
|
| D | gpu.Dockerfile | 26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
38 cuda-command-line-tools-${CUDA/./-} \
39 cuda-cublas-${CUDA/./-} \
40 cuda-cufft-${CUDA/./-} \
41 cuda-curand-${CUDA/./-} \
42 cuda-cusolver-${CUDA/./-} \
43 cuda-cusparse-${CUDA/./-} \
45 libcudnn7=${CUDNN}+cuda${CUDA} \
54 apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
56 && apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda${CUDA} \
[all …]
|
| D | devel-gpu-jupyter.Dockerfile | 26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
39 cuda-command-line-tools-${CUDA/./-} \
40 cuda-cublas-dev-${CUDA/./-} \
41 cuda-cudart-dev-${CUDA/./-} \
42 cuda-cufft-dev-${CUDA/./-} \
43 cuda-curand-dev-${CUDA/./-} \
44 cuda-cusolver-dev-${CUDA/./-} \
45 cuda-cusparse-dev-${CUDA/./-} \
46 libcudnn7=${CUDNN}+cuda${CUDA} \
47 libcudnn7-dev=${CUDNN}+cuda${CUDA} \
[all …]
|
| D | gpu-jupyter.Dockerfile | 26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
38 cuda-command-line-tools-${CUDA/./-} \
39 cuda-cublas-${CUDA/./-} \
40 cuda-cufft-${CUDA/./-} \
41 cuda-curand-${CUDA/./-} \
42 cuda-cusolver-${CUDA/./-} \
43 cuda-cusparse-${CUDA/./-} \
45 libcudnn7=${CUDNN}+cuda${CUDA} \
54 apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
56 && apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda${CUDA} \
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | BUILD | 15 load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda_is_configured")
82 "@local_config_cuda//cuda:cuda_headers",
98 "@local_config_cuda//cuda:cuda_headers",
108 "//conditions:default": ["//tensorflow/core:cuda"],
124 "@local_config_cuda//cuda:cuda_headers",
130 "@local_config_cuda//cuda:cuda_headers",
156 "@local_config_cuda//cuda:cuda_headers",
182 "@local_config_cuda//cuda:cuda_headers",
202 "@local_config_cuda//cuda:cuda_headers",
214 ["@local_config_cuda//cuda:cublas"],
[all …]
|
| D | cuda_diagnostics.cc | 53 namespace cuda { namespace
198 << cuda::DriverVersionStatusToString(dso_version); in LogDriverVersionInformation()
202 << cuda::DriverVersionStatusToString(kernel_version); in LogDriverVersionInformation()
240 result = cuda::StringToDriverVersion(version); in FindDsoVersion()
268 *result = cuda::StringToDriverVersion(stripped_dso_version); in FindDsoVersion()
300 return cuda::StringToDriverVersion(stripped_kernel_version); in FindKernelModuleVersion()
309 << cuda::DriverVersionToString(kernel_version.ValueOrDie()); in WarnOnDsoKernelMismatch()
312 << cuda::DriverVersionStatusToString(kernel_version) in WarnOnDsoKernelMismatch()
314 << cuda::DriverVersionStatusToString(dso_version) in WarnOnDsoKernelMismatch()
344 return cuda::StringToDriverVersion(""); in FindKernelDriverVersion()
[all …]
|
| D | cuda_rng.cc | 68 cuda::ScopedActivateExecutorContext sac(parent_); in ~GpuRng()
77 cuda::ScopedActivateExecutorContext sac(parent_); in Init()
89 cuda::ScopedActivateExecutorContext sac(parent_); in SetStream()
124 cuda::ScopedActivateExecutorContext sac(parent_); in DoPopulateRandUniformInternal()
173 cuda::ScopedActivateExecutorContext sac(parent_); in DoPopulateRandGaussianInternal()
211 cuda::ScopedActivateExecutorContext sac(parent_); in SetSeed()
234 cuda::kCudaPlatformId, gpu::kGpuRandPlugin, "cuRAND", in initialize_curand()
260 cuda::kCudaPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_curand()
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | tf_stream_executor.cmake | 27 # "cuda/*.cc",
32 # "cuda/*.h",
37 # "//tensorflow/core:cuda",
38 # "//third_party/gpus/cuda:cublas",
39 # "//third_party/gpus/cuda:cudnn",
47 # "//third_party/gpus/cuda:cuda_headers",
66 "${tensorflow_source_dir}/tensorflow/stream_executor/cuda/*.cc"
70 "${tensorflow_source_dir}/tensorflow/stream_executor/cuda/*_test.cc"
|
| /external/tensorflow/tensorflow/tools/dockerfiles/tests/ |
| D | 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/tensorflow/tensorflow/contrib/makefile/ |
| D | 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
|