Home
last modified time | relevance | path

Searched refs:cuda (Results 1 – 25 of 387) sorted by relevance

12345678910>>...16

/external/tensorflow/third_party/gpus/cuda/
DBUILD.tpl44 "cuda/cuda_config.h",
45 ":cuda-include"
49 ".", # required to include cuda/cuda/cuda_config.h as cuda/config.h
50 "cuda/include",
56 srcs = ["cuda/lib/%{cudart_static_lib}"],
68 srcs = ["cuda/lib/%{cuda_driver_lib}"],
73 srcs = ["cuda/lib/%{cudart_lib}"],
74 data = ["cuda/lib/%{cudart_lib}"],
81 include_prefix = "third_party/gpus/cuda/include",
90 include_prefix = "third_party/gpus/cuda/include",
[all …]
DBUILD.windows.tpl41 # Provides CUDA headers for '#include "third_party/gpus/cuda/include/cuda.h"'
46 "cuda/cuda_config.h",
47 ":cuda-include"
51 ".", # required to include cuda/cuda/cuda_config.h as cuda/config.h
52 "cuda/include",
63 interface_library = "cuda/lib/%{cudart_static_lib}",
69 interface_library = "cuda/lib/%{cuda_driver_lib}",
75 interface_library = "cuda/lib/%{cudart_lib}",
82 include_prefix = "third_party/gpus/cuda/include",
91 include_prefix = "third_party/gpus/cuda/include",
[all …]
Dbuild_defs.bzl.tpl10 "@local_config_cuda//cuda:using_nvcc": if_true,
11 "@local_config_cuda//cuda:using_clang": if_true,
16 """Shorthand for select()'ing on wheteher we're building with cuda-clang.
19 with cuda-clang. Otherwise, the select statement evaluates to if_false.
23 "@local_config_cuda//cuda:using_clang": if_true,
28 """Shorthand for select()'ing on wheteher we're building with cuda-clang
32 with cuda-clang in opt mode. Otherwise, the select statement evaluates to
37 "@local_config_cuda//cuda:using_clang_opt": if_true,
44 "-x", "cuda",
47 "--no-cuda-include-ptx=all"
[all …]
/external/llvm-project/clang/test/Driver/
Dhip-output-file-name.hip6 // RUN: --cuda-gpu-arch=gfx803 --cuda-gpu-arch=gfx900 %s \
14 // RUN: --cuda-gpu-arch=gfx803 --cuda-gpu-arch=gfx900 %s \
18 // RUN: --cuda-gpu-arch=gfx803 --cuda-gpu-arch=gfx900 %s \
21 // RUN: %clang -### -E --cuda-device-only -target x86_64-linux-gnu \
22 // RUN: --cuda-gpu-arch=gfx803 --cuda-gpu-arch=gfx900 %s \
25 // RUN: %clang -### -E --cuda-host-only -target x86_64-linux-gnu \
26 // RUN: --cuda-gpu-arch=gfx803 --cuda-gpu-arch=gfx900 %s \
35 // RUN: --cuda-gpu-arch=gfx803 --cuda-gpu-arch=gfx900 %s \
39 // RUN: --cuda-gpu-arch=gfx803 --cuda-gpu-arch=gfx900 %s \
42 // RUN: %clang -### -E -o test.cui --cuda-device-only -target x86_64-linux-gnu \
[all …]
Dhip-device-libs.hip10 // RUN: --cuda-gpu-arch=gfx803 \
18 // RUN: --cuda-gpu-arch=gfx900 \
26 // RUN: --cuda-gpu-arch=gfx900 \
35 // RUN: --cuda-gpu-arch=gfx803 \
36 // RUN: -fno-cuda-flush-denormals-to-zero \
44 // RUN: --cuda-gpu-arch=gfx900 \
45 // RUN: -fno-cuda-flush-denormals-to-zero \
53 // RUN: --cuda-gpu-arch=gfx803 \
62 // RUN: --cuda-gpu-arch=gfx803 \
63 // RUN: -fcuda-flush-denormals-to-zero -fno-cuda-flush-denormals-to-zero \
[all …]
Dhip-toolchain-features.hip6 // RUN: --cuda-gpu-arch=gfx906:xnack+ --cuda-gpu-arch=gfx900:xnack+ %s \
9 // RUN: --cuda-gpu-arch=gfx906:xnack- --cuda-gpu-arch=gfx900:xnack- %s \
18 // RUN: --cuda-gpu-arch=gfx908:sramecc+ %s \
21 // RUN: --cuda-gpu-arch=gfx908:sramecc- %s \
30 // RUN: --cuda-gpu-arch=gfx1010 %s \
33 // RUN: --cuda-gpu-arch=gfx1010 %s \
42 // RUN: --cuda-gpu-arch=gfx908:xnack+:sramecc+ %s \
45 // RUN: --cuda-gpu-arch=gfx908:xnack-:sramecc- %s \
52 // RUN: --cuda-gpu-arch=gfx1010 %s \
63 // RUN: --cuda-gpu-arch=gfx1010 %s \
Dhip-wavefront-size.hip4 // RUN: --cuda-gpu-arch=gfx900 \
5 // RUN: --rocm-path=%S/Inputs/rocm --cuda-device-only %s \
10 // RUN: --cuda-gpu-arch=gfx1010 \
11 // RUN: --rocm-path=%S/Inputs/rocm --cuda-device-only %s \
16 // RUN: --cuda-gpu-arch=gfx1010 \
17 // RUN: --cuda-gpu-arch=gfx900 \
18 // RUN: --rocm-path=%S/Inputs/rocm --cuda-device-only %s \
Dhip-device-compile.hip8 // RUN: %clang -c -emit-llvm --cuda-device-only -### -target x86_64-linux-gnu \
9 // RUN: -o a.bc -x hip --cuda-gpu-arch=gfx900 \
15 // RUN: %clang -c -S -emit-llvm --cuda-device-only -### -target x86_64-linux-gnu \
16 // RUN: -o a.ll -x hip --cuda-gpu-arch=gfx900 \
22 // RUN: %clang -c -S --cuda-device-only -### -target x86_64-linux-gnu \
23 // RUN: -o a.s -x hip --cuda-gpu-arch=gfx900 \
53 // RUN: %clang -c --cuda-device-only -### -target x86_64-linux-gnu \
54 // RUN: -o a.s -x hip --cuda-gpu-arch=gfx900 \
60 // RUN: %clang --cuda-device-only -### -target x86_64-linux-gnu \
61 // RUN: -o a.s -x hip --cuda-gpu-arch=gfx900 \
/external/tensorflow/tensorflow/tools/dockerfiles/partials/ubuntu/
Ddevel-nvidia.partial.Dockerfile3 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
18 cuda-command-line-tools-${CUDA/./-} \
21 cuda-nvprune-${CUDA/./-} \
22 cuda-nvrtc-${CUDA/./-} \
23 cuda-nvrtc-dev-${CUDA/./-} \
24 cuda-cudart-dev-${CUDA/./-} \
29 libcudnn8=${CUDNN}+cuda${CUDA} \
30 libcudnn8-dev=${CUDNN}+cuda${CUDA} \
44 …find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' …
49 …nstall -y --no-install-recommends libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
[all …]
Dnvidia.partial.Dockerfile3 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
19 cuda-command-line-tools-${CUDA/./-} \
21 cuda-nvrtc-${CUDA/./-} \
27 libcudnn8=${CUDNN}+cuda${CUDA} \
37 …nstall -y --no-install-recommends libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
38 libnvinfer-plugin${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
43 ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
47 RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 \
48 && echo "/usr/local/cuda/lib64/stubs" > /etc/ld.so.conf.d/z-cuda-stubs.conf \
/external/llvm-project/mlir/tools/mlir-cuda-runner/
DCMakeLists.txt2 cuda-runtime-wrappers.cpp
3 mlir-cuda-runner.cpp
13 "Building the mlir cuda runner requires the NVPTX backend")
24 "Building the mlir cuda runner requires a working CUDA install")
28 find_library(CUDA_RUNTIME_LIBRARY cuda)
30 add_llvm_library(cuda-runtime-wrappers SHARED
31 cuda-runtime-wrappers.cpp
33 target_include_directories(cuda-runtime-wrappers
37 target_link_libraries(cuda-runtime-wrappers
73 add_llvm_tool(mlir-cuda-runner
[all …]
/external/tensorflow/tensorflow/core/platform/default/build_config/
DBUILD33 "//tensorflow/stream_executor/cuda:cuda_activation",
42 "//tensorflow/stream_executor/cuda:cuda_platform_id",
47 "@local_config_cuda//cuda:darwin": ["IOKit"],
50 "//tensorflow:using_cuda_clang": ["//tensorflow/stream_executor/cuda:all_runtime"],
51 "//tensorflow:using_cuda_nvcc": ["//tensorflow/stream_executor/cuda:all_runtime"],
64 "//tensorflow:oss": ["//tensorflow/stream_executor/cuda:cudart_stub"],
66 "//tensorflow/stream_executor/cuda:all_runtime",
67 ":cuda",
70 "@local_config_cuda//cuda:darwin": ["IOKit"],
95 "//tensorflow/stream_executor/cuda:cuda_platform_id",
[all …]
/external/skqp/src/compute/hs/cuda/bench/
Dmain.c293 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/ci_build/
DDockerfile.rbe.cuda9.0-cudnn7-ubuntu14.0416 …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 …]
DDockerfile.rbe.cuda10.0-cudnn7-ubuntu14.0413 …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 …]
DDockerfile.gpu1 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
DDockerfile.gpu.ppc64le1 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/dockerfiles/dockerfiles/ppc64le/
Ddevel-gpu-ppc64le.Dockerfile26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
41 cuda-command-line-tools-${CUDA/./-} \
44 cuda-nvprune-${CUDA/./-} \
45 cuda-nvrtc-${CUDA/./-} \
46 cuda-nvrtc-dev-${CUDA/./-} \
47 cuda-cudart-dev-${CUDA/./-} \
52 libcudnn8=${CUDNN}+cuda${CUDA} \
53 libcudnn8-dev=${CUDNN}+cuda${CUDA} \
67 …find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' …
72 …nstall -y --no-install-recommends libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
[all …]
Ddevel-gpu-ppc64le-jupyter.Dockerfile26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
41 cuda-command-line-tools-${CUDA/./-} \
44 cuda-nvprune-${CUDA/./-} \
45 cuda-nvrtc-${CUDA/./-} \
46 cuda-nvrtc-dev-${CUDA/./-} \
47 cuda-cudart-dev-${CUDA/./-} \
52 libcudnn8=${CUDNN}+cuda${CUDA} \
53 libcudnn8-dev=${CUDNN}+cuda${CUDA} \
67 …find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' …
72 …nstall -y --no-install-recommends libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
[all …]
/external/tensorflow/tensorflow/tools/dockerfiles/dockerfiles/
Ddevel-gpu.Dockerfile26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
41 cuda-command-line-tools-${CUDA/./-} \
44 cuda-nvprune-${CUDA/./-} \
45 cuda-nvrtc-${CUDA/./-} \
46 cuda-nvrtc-dev-${CUDA/./-} \
47 cuda-cudart-dev-${CUDA/./-} \
52 libcudnn8=${CUDNN}+cuda${CUDA} \
53 libcudnn8-dev=${CUDNN}+cuda${CUDA} \
67 …find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' …
72 …nstall -y --no-install-recommends libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
[all …]
Ddevel-gpu-jupyter.Dockerfile26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
41 cuda-command-line-tools-${CUDA/./-} \
44 cuda-nvprune-${CUDA/./-} \
45 cuda-nvrtc-${CUDA/./-} \
46 cuda-nvrtc-dev-${CUDA/./-} \
47 cuda-cudart-dev-${CUDA/./-} \
52 libcudnn8=${CUDNN}+cuda${CUDA} \
53 libcudnn8-dev=${CUDNN}+cuda${CUDA} \
67 …find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' …
72 …nstall -y --no-install-recommends libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
[all …]
Dgpu.Dockerfile26 FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base
42 cuda-command-line-tools-${CUDA/./-} \
44 cuda-nvrtc-${CUDA/./-} \
50 libcudnn8=${CUDNN}+cuda${CUDA} \
60 …nstall -y --no-install-recommends libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
61 libnvinfer-plugin${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
66 ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
70 RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 \
71 && echo "/usr/local/cuda/lib64/stubs" > /etc/ld.so.conf.d/z-cuda-stubs.conf \
/external/fmtlib/test/cuda-test/
DCMakeLists.txt13 # https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#cpp14-language-features
23 # https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html
32 cuda_add_executable(fmt-in-cuda-test cuda-cpp14.cu cpp14.cc)
33 target_compile_features(fmt-in-cuda-test PRIVATE cxx_std_14)
41 target_compile_options(fmt-in-cuda-test PRIVATE /Zc:__cplusplus /permissive-)
45 add_executable(fmt-in-cuda-test cuda-cpp14.cu cpp14.cc)
46 set_target_properties(fmt-in-cuda-test PROPERTIES CUDA_SEPARABLE_COMPILATION ON)
47 target_compile_features(fmt-in-cuda-test PRIVATE cxx_std_14)
55 set_property(SOURCE cuda-cpp14.cu APPEND PROPERTY
62 get_target_property(IN_USE_CUDA_STANDARD fmt-in-cuda-test CUDA_STANDARD)
[all …]
/external/tensorflow/third_party/gpus/
Dcuda_configure.bzl7 * `TF_CUDA_CLANG`: Whether to use clang as a cuda compiler.
15 `/usr/local/cuda,usr/`.
17 `/usr/local/cuda`.
22 `/usr/local/cuda`.
326 """Output failure message when cuda configuration fails."""
334 """Generates the Starlark string with cuda include directories.
412 """Returns a list of strings representing cuda compute capabilities.
416 Returns: list of cuda architectures to compile for. 'compute_xy' refers to
515 check_cuda_libs_script: The path to a script verifying that the cuda
526 "cuda": _check_cuda_lib_params(
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
DBUILD99 "@local_config_cuda//cuda:cuda_headers",
123 "@local_config_cuda//cuda:cuda_headers",
133 "//conditions:default": ["//tensorflow/core:cuda"],
152 "@local_config_cuda//cuda:cuda_headers",
182 "@local_config_cuda//cuda:cuda_headers",
188 "@local_config_cuda//cuda:cuda_headers",
214 "@local_config_cuda//cuda:cuda_headers",
241 "@local_config_cuda//cuda:cublas_headers",
243 "@local_config_cuda//cuda:cuda_headers",
253 "//conditions:default": "@local_config_cuda//cuda:cublas",
[all …]

12345678910>>...16