drivers/gpu/cuda/common.h

   1 /* SPDX-License-Identifier: BSD-3-Clause
   2  * Copyright (c) 2022 NVIDIA Corporation & Affiliates
   3  */
   4
   5 #ifndef CUDA_COMMON_H
   6 #define CUDA_COMMON_H
   7
   8 #include <dlfcn.h>
   9
  10 #include <rte_common.h>
  11 #include <rte_log.h>
  12 #include <rte_errno.h>
  13
  14 extern int cuda_logtype;
  15
  16 /* Helper macro for logging */
  17 #define rte_cuda_log(level, fmt, ...) \
  18         rte_log(RTE_LOG_ ## level, cuda_logtype, fmt "\n", ##__VA_ARGS__)
  19
  20 #define rte_cuda_debug(fmt, ...) \
  21         rte_cuda_log(DEBUG, RTE_STR(__LINE__) ":%s() " fmt, __func__, \
  22                 ##__VA_ARGS__)
  23
  24 #ifdef DRIVERS_GPU_CUDA_GDRCOPY_H
  25         #include <gdrapi.h>
  26 #else
  27         struct gdr;
  28         typedef struct gdr *gdr_t;
  29         struct gdr_mh_s { unsigned long h; };
  30         typedef struct gdr_mh_s gdr_mh_t;
  31 #endif
  32
  33 int gdrcopy_pin(gdr_t *gdrc_h, __rte_unused gdr_mh_t *mh,
  34                 uint64_t d_addr, size_t size, void **h_addr);
  35 int gdrcopy_unpin(gdr_t gdrc_h, __rte_unused gdr_mh_t mh,
  36                 void *d_addr, size_t size);
  37
  38 #endif /* CUDA_COMMON_H */