nvidia.yml 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. # build/platform/cuda (include/ and extras/CUPTI/include/)
  2. - includes:
  3. - cooperative_groups.h
  4. - cuComplex.h
  5. # Though cub is provided as a part of CUDA HPC SDK,
  6. # we would like to be updatable independently.
  7. # Hence we are using it directly from Arcadia.
  8. # This sysincl enforces such usage.
  9. - cub/cub.cuh: contrib/libs/nvidia/cub/cub/cub.cuh
  10. - cublas.h
  11. - cublasLt.h
  12. - cublas_v2.h
  13. - cuda.h
  14. - cuda_bf16.h
  15. - cuda_fp16.h
  16. - cuda_gl_interop.h
  17. - cuda_runtime.h
  18. - cuda_runtime_api.h
  19. - cufft.h
  20. - cupti.h
  21. - curand.h
  22. - curand_kernel.h
  23. - curand_mtgp32_host.h
  24. - curand_mtgp32dc_p_11213.h
  25. - cusolverDn.h
  26. - cusparse.h
  27. - driver_types.h
  28. - host_defines.h
  29. - library_types.h
  30. - math_constants.h
  31. - npp.h
  32. - nppi.h
  33. - nvToolsExt.h
  34. # Though thrust is provided as a part of CUDA HPC SDK,
  35. # we would like to be updatable independently.
  36. # Hence we are using it directly from Arcadia.
  37. # This sysincl enforces such usage.
  38. - thrust/iterator/iterator_traits.h: contrib/libs/nvidia/thrust/thrust/iterator/iterator_traits.h
  39. - thrust/iterator/iterator_facade.h: contrib/libs/nvidia/thrust/thrust/iterator/iterator_facade.h
  40. - thrust/system/cuda/detail/core/triple_chevron_launch.h: contrib/libs/nvidia/thrust/thrust/system/cuda/detail/core/triple_chevron_launch.h
  41. - thrust/version.h: contrib/libs/nvidia/thrust/thrust/version.h
  42. - vector_types.h
  43. # These headers should be provided by TensorRT.
  44. # Proper solution must be found in DTCC-171
  45. - source_filter: "^contrib/libs/tf-2.4/.*"
  46. includes:
  47. - NvInfer.h
  48. - NvInferPlugin.h