Go to the documentation of this file.
8 #include <unordered_map>
45 #if defined(YAKL_ARCH_CUDA)
46 #include <cub/cub.cuh>
48 #elif defined(YAKL_ARCH_HIP)
49 #include "hip/hip_runtime.h"
50 #include "hipcub/hipcub.hpp"
51 #include "roctracer/roctx.h"
53 #elif defined(YAKL_ARCH_SYCL)
54 #include <sycl/sycl.hpp>
55 #include <oneapi/mkl/exceptions.hpp>
56 #include <oneapi/mkl/dfti.hpp>
57 #elif defined(YAKL_ARCH_OPENMP)
62 #include <nvtx3/nvToolsExt.h>