nvidia.yml 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. # build/platform/cuda (include/ and extras/CUPTI/include/)
  2. - includes:
  3. - cooperative_groups.h
  4. - cooperative_groups/reduce.h
  5. - cuComplex.h
  6. - cublas.h
  7. - cublasLt.h
  8. - cublas_v2.h
  9. - cuda.h
  10. - cuda/std/cassert
  11. - cuda/std/cmath
  12. - cuda/std/cstddef
  13. - cuda/std/cstdint
  14. - cuda/std/functional
  15. - cuda/std/limits
  16. - cuda/std/tuple
  17. - cuda/std/type_traits
  18. - cuda/std/utility
  19. - cudaEGL.h
  20. - cudaProfiler.h
  21. - cudaTypedefs.h
  22. - cuda_bf16.h
  23. - cuda_fp16.h
  24. - cuda_fp8.h
  25. - cuda_gl_interop.h
  26. - cuda_occupancy.h
  27. - cuda_profiler_api.h
  28. - cuda_runtime.h
  29. - cuda_runtime_api.h
  30. - cufft.h
  31. - cufftXt.h
  32. - cupti.h
  33. - cupti_profiler_target.h
  34. - cupti_target.h
  35. - curand.h
  36. - curand_kernel.h
  37. - curand_mtgp32_host.h
  38. - curand_mtgp32dc_p_11213.h
  39. - cusolverDn.h
  40. - cusolverSp.h
  41. - cusparse.h
  42. - cusparseLt.h
  43. - device_atomic_functions.h
  44. - device_types.h
  45. - driver_types.h
  46. - host_defines.h
  47. - library_types.h
  48. - math_constants.h
  49. - mma.h
  50. - npp.h
  51. - nppi.h
  52. - nv/target
  53. - nvToolsExt.h
  54. - nvToolsExtCuda.h
  55. - nvml.h
  56. - nvperf_cuda_host.h
  57. - nvperf_host.h
  58. - nvperf_target.h
  59. - nvrtc.h
  60. - surface_types.h
  61. - texture_types.h
  62. - vector_types.h
  63. # These headers should be provided by TensorRT.
  64. # Proper solution must be found in DTCC-171
  65. - source_filter: "^contrib/libs/tf/.*"
  66. includes:
  67. - NvInfer.h
  68. - NvInferPlugin.h