tree: 58a4edbb477b2efab3d2ece96fb188f7b71b8091 [path history] [tgz]
  1. BUILD.template
  2. CMakeLists.txt
  3. README.md
build_tools/third_party/cuda/README.md

CUDA Toolkit Configuration

This directory contains:

  • CMake support for auto-downloading and finding the CUDA toolkit.
  • Targets in an “iree_cuda” namespace that encapsulate CUDA for the rest of the codebase.
  • Bazel overlay.

Note that the amount of CUDA that IREE itself depends on is minimal and encapsulated to targets in this directory (versus being loose leaf across the codebase).