blob: 17a7bb1045e25618b9895774e4dbc5339ff9b17f (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
# build/platform/cuda (include/ and extras/CUPTI/include/)
- includes:
- cooperative_groups.h
- cooperative_groups/reduce.h
- cuComplex.h
# Though cub is provided as a part of CUDA HPC SDK,
# we would like to be updatable independently.
# Hence we are using it directly from Arcadia.
# This sysincl enforces such usage.
- cub/cub.cuh: contrib/libs/nvidia/cub/cub/cub.cuh
- cublas.h
- cublasLt.h
- cublas_v2.h
- cuda.h
- cudaProfiler.h
- cudaTypedefs.h
- cuda_bf16.h
- cuda_fp16.h
- cuda_gl_interop.h
- cuda_occupancy.h
- cuda_profiler_api.h
- cuda_runtime.h
- cuda_runtime_api.h
- cufft.h
- cufftXt.h
- cupti.h
- curand.h
- curand_kernel.h
- curand_mtgp32_host.h
- curand_mtgp32dc_p_11213.h
- cusolverDn.h
- cusolverSp.h
- cusparse.h
- cusparseLt.h
- device_atomic_functions.h
- device_types.h
- driver_types.h
- host_defines.h
- library_types.h
- math_constants.h
- nvrtc.h
- nvToolsExtCuda.h
- npp.h
- nppi.h
- nvToolsExt.h
- nvml.h
- surface_types.h
- texture_types.h
- vector_types.h
# Though thrust is provided as a part of CUDA HPC SDK,
# we would like to be updatable independently.
# Hence we are using it directly from Arcadia.
# This sysincl enforces such usage.
- thrust/iterator/iterator_traits.h: contrib/libs/nvidia/thrust/thrust/iterator/iterator_traits.h
- thrust/iterator/iterator_facade.h: contrib/libs/nvidia/thrust/thrust/iterator/iterator_facade.h
- thrust/system/cuda/detail/core/triple_chevron_launch.h: contrib/libs/nvidia/thrust/thrust/system/cuda/detail/core/triple_chevron_launch.h
- thrust/version.h: contrib/libs/nvidia/thrust/thrust/version.h
# These headers should be provided by TensorRT.
# Proper solution must be found in DTCC-171
- source_filter: "^contrib/libs/tf/.*"
includes:
- NvInfer.h
- NvInferPlugin.h
# Add support nvrtc
- source_filter: "^contrib/libs/nvidia/cutlass/.*"
includes:
- cuda/std/cassert
- cuda/std/cmath
- cuda/std/cstddef
- cuda/std/cstdint
- cuda/std/limits
- cuda/std/tuple
- cuda/std/type_traits
- cuda/std/utility
- cuda_fp8.h
- mma.h
|
