| def define_targets(rules): | |
| rules.cc_library( | |
| name = "c10", | |
| visibility = ["//visibility:public"], | |
| deps = [ | |
| "//c10/core:CPUAllocator", | |
| "//c10/core:ScalarType", | |
| "//c10/core:alignment", | |
| "//c10/core:alloc_cpu", | |
| "//c10/core:base", | |
| "//c10/macros", | |
| "//c10/mobile:CPUCachingAllocator", | |
| "//c10/mobile:CPUProfilingAllocator", | |
| "//c10/util:TypeCast", | |
| "//c10/util:base", | |
| "//c10/util:typeid", | |
| ] + rules.if_cuda( | |
| [ | |
| "//c10/cuda:cuda", | |
| "//c10/cuda:Macros", | |
| ], | |
| [], | |
| ), | |
| ) |