Edward Yang | ce7910b | 2022-02-28 15:46:04 -0800 | [diff] [blame] | 1 | # WARNING: the contents of this file must BOTH be valid Starlark (for Buck and |
| 2 | |
| 3 | # Bazel) as well as valid Python (for our cmake build). This means that |
| 4 | # load() directives are not allowed (as they are not recognized by Python). |
| 5 | # If you want to fix this, figure out how run this file from cmake with a proper |
| 6 | # Starlark interpreter as part of the default OSS build process. If you need |
| 7 | # some nontrivial Starlark features, make a separate bzl file (remember that |
| 8 | |
| 9 | # bzl files are not exported via ShipIt by default, so you may also need to |
| 10 | # update PyTorch's ShipIt config) |
| 11 | |
Will Constable | 3547f20 | 2022-03-22 16:06:04 -0700 | [diff] [blame] | 12 | # This is duplicated in caffe2/CMakeLists.txt for now and not yet used in buck |
| 13 | GENERATED_LAZY_TS_CPP = [ |
| 14 | "lazy/generated/LazyNativeFunctions.cpp", |
| 15 | "lazy/generated/RegisterAutogradLazy.cpp", |
| 16 | "lazy/generated/RegisterLazy.cpp", |
| 17 | ] |
| 18 | |
Nikita Shulga | 1571dd8 | 2020-06-30 13:42:01 -0700 | [diff] [blame] | 19 | def libtorch_generated_sources(gencode_pattern): |
| 20 | return [gencode_pattern.format(name) for name in [ |
mikey dagitses | 096ff0e | 2022-05-04 06:48:20 -0700 | [diff] [blame] | 21 | "torch/csrc/autograd/generated/Functions.cpp", |
| 22 | "torch/csrc/autograd/generated/VariableType_0.cpp", |
| 23 | "torch/csrc/autograd/generated/VariableType_1.cpp", |
| 24 | "torch/csrc/autograd/generated/VariableType_2.cpp", |
| 25 | "torch/csrc/autograd/generated/VariableType_3.cpp", |
| 26 | "torch/csrc/autograd/generated/VariableType_4.cpp", |
| 27 | "torch/csrc/autograd/generated/TraceType_0.cpp", |
| 28 | "torch/csrc/autograd/generated/TraceType_1.cpp", |
| 29 | "torch/csrc/autograd/generated/TraceType_2.cpp", |
| 30 | "torch/csrc/autograd/generated/TraceType_3.cpp", |
| 31 | "torch/csrc/autograd/generated/TraceType_4.cpp", |
| 32 | "torch/csrc/autograd/generated/ADInplaceOrViewType_0.cpp", |
| 33 | "torch/csrc/autograd/generated/ADInplaceOrViewType_1.cpp", |
Michael Ranieri | 689499f | 2020-10-09 20:15:14 -0700 | [diff] [blame] | 34 | ]] |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 35 | |
Sam Estep | 707d271 | 2020-10-29 18:42:03 -0700 | [diff] [blame] | 36 | # copied from https://github.com/pytorch/pytorch/blob/f99a693cd9ff7a9b5fdc71357dac66b8192786d3/aten/src/ATen/core/CMakeLists.txt |
Nikita Shulga | f3d2e33 | 2020-05-13 22:15:07 -0700 | [diff] [blame] | 37 | jit_core_headers = [ |
Peter Bell | cd9da32 | 2021-12-07 15:22:44 -0800 | [diff] [blame] | 38 | "torch/csrc/Export.h", |
Nikita Shulga | f3d2e33 | 2020-05-13 22:15:07 -0700 | [diff] [blame] | 39 | "torch/csrc/jit/frontend/source_range.h", |
Kimish Patel | e0fc473 | 2021-05-04 09:17:43 -0700 | [diff] [blame] | 40 | "torch/csrc/jit/serialization/callstack_debug_info_serialization.h", |
Nikita Shulga | f3d2e33 | 2020-05-13 22:15:07 -0700 | [diff] [blame] | 41 | "torch/csrc/jit/serialization/source_range_serialization.h", |
| 42 | "torch/csrc/jit/frontend/lexer.h", |
| 43 | "torch/csrc/jit/frontend/strtod.h", |
| 44 | "torch/csrc/jit/frontend/parser_constants.h", |
| 45 | "torch/csrc/jit/frontend/function_schema_parser.h", |
| 46 | "torch/csrc/jit/frontend/parse_string_literal.h", |
| 47 | "torch/csrc/jit/frontend/schema_type_parser.h", |
| 48 | "torch/csrc/jit/frontend/error_report.h", |
| 49 | "torch/csrc/jit/frontend/tree.h", |
| 50 | "torch/custom_class.h", |
| 51 | "torch/custom_class_detail.h", |
| 52 | "torch/library.h", |
| 53 | ] |
| 54 | |
| 55 | jit_core_sources = [ |
| 56 | "torch/csrc/jit/frontend/error_report.cpp", |
| 57 | "torch/csrc/jit/frontend/function_schema_parser.cpp", |
| 58 | "torch/csrc/jit/frontend/lexer.cpp", |
| 59 | "torch/csrc/jit/frontend/schema_type_parser.cpp", |
| 60 | "torch/csrc/jit/frontend/strtod.cpp", |
| 61 | "torch/csrc/jit/frontend/source_range.cpp", |
| 62 | ] |
| 63 | |
Sam Estep | 707d271 | 2020-10-29 18:42:03 -0700 | [diff] [blame] | 64 | # copied from https://github.com/pytorch/pytorch/blob/0bde610c14b92d351b968a0228df29e92442b1cc/torch/CMakeLists.txt |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 65 | # There are some common files used in both internal lite-interpreter and full-jit. Making a separate |
| 66 | # list for the shared files. |
| 67 | |
| 68 | core_sources_common = [ |
albanD | c23808d | 2020-12-22 12:07:00 -0800 | [diff] [blame] | 69 | "torch/csrc/autograd/autograd_meta.cpp", |
| 70 | "torch/csrc/autograd/forward_grad.cpp", |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 71 | "torch/csrc/jit/frontend/edit_distance.cpp", |
Jacob Szwejbka | e926360 | 2022-01-20 10:04:43 -0800 | [diff] [blame] | 72 | "torch/csrc/jit/mobile/compatibility/runtime_compatibility.cpp", |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 73 | "torch/csrc/jit/mobile/type_parser.cpp", |
Tugsbayasgalan (Tugsuu) Manlaibaatar | e849c8b | 2022-01-27 18:27:28 -0800 | [diff] [blame] | 74 | "torch/csrc/jit/operator_upgraders/version_map.cpp", |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 75 | "torch/csrc/jit/runtime/instruction.cpp", |
| 76 | "torch/csrc/jit/runtime/jit_exception.cpp", |
| 77 | "torch/csrc/jit/runtime/operator.cpp", |
Pavithran Ramachandran | bf69a61 | 2022-01-26 18:51:40 -0800 | [diff] [blame] | 78 | "torch/csrc/jit/mobile/register_ops_common_utils.cpp", |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 79 | "torch/csrc/jit/runtime/print_handler.cpp", |
Tugsbayasgalan Manlaibaatar | 42c78ed | 2020-12-11 10:58:54 -0800 | [diff] [blame] | 80 | "torch/csrc/jit/runtime/slice_indices_adjust.cpp", |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 81 | "torch/csrc/jit/runtime/register_ops_utils.cpp", |
| 82 | "torch/csrc/jit/runtime/vararg_functions.cpp", |
Pavithran Ramachandran | 3c003aa | 2021-09-17 10:22:41 -0700 | [diff] [blame] | 83 | "torch/csrc/jit/mobile/promoted_prim_ops.cpp", |
| 84 | "torch/csrc/jit/mobile/prim_ops_registery.cpp", |
Taylor Robie | ebc66bf | 2021-12-16 10:32:13 -0800 | [diff] [blame] | 85 | "torch/csrc/profiler/util.cpp", |
Pavithran Ramachandran | c2ceba8 | 2021-11-04 15:52:03 -0700 | [diff] [blame] | 86 | ] |
| 87 | |
| 88 | torch_unpickler_common = [ |
Chen Lai | 9486fc3 | 2021-04-29 10:06:38 -0700 | [diff] [blame] | 89 | "torch/csrc/jit/serialization/import_read.cpp", |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 90 | "torch/csrc/jit/serialization/unpickler.cpp", |
| 91 | ] |
| 92 | |
Pavithran Ramachandran | c2ceba8 | 2021-11-04 15:52:03 -0700 | [diff] [blame] | 93 | libtorch_sources_common = sorted(core_sources_common + torch_unpickler_common) |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 94 | |
Dhruv Matani | 8c852de | 2021-06-06 13:14:28 -0700 | [diff] [blame] | 95 | # The profilers are not needed in the lite interpreter build. |
| 96 | libtorch_profiler_sources = [ |
PyTorch MergeBot | 8eb579e | 2022-10-14 14:56:59 +0000 | [diff] [blame] | 97 | "torch/csrc/autograd/profiler_legacy.cpp", |
Dhruv Matani | 8c852de | 2021-06-06 13:14:28 -0700 | [diff] [blame] | 98 | "torch/csrc/autograd/profiler_kineto.cpp", |
Taylor Robie | daf959c | 2022-03-16 10:20:44 -0700 | [diff] [blame] | 99 | "torch/csrc/profiler/collection.cpp", |
Taylor Robie | fb64f7b | 2022-10-26 16:56:51 -0700 | [diff] [blame] | 100 | "torch/csrc/profiler/data_flow.cpp", |
Taylor Robie | 786f946 | 2022-01-11 15:56:17 -0800 | [diff] [blame] | 101 | "torch/csrc/profiler/kineto_shim.cpp", |
Jay Chae | 77665e9 | 2022-04-20 16:56:39 -0700 | [diff] [blame] | 102 | "torch/csrc/profiler/kineto_client_interface.cpp", |
Taylor Robie | bea0184 | 2022-09-07 14:58:28 -0700 | [diff] [blame] | 103 | "torch/csrc/profiler/orchestration/observer.cpp", |
Taylor Robie | 3285387 | 2022-09-08 11:03:20 -0700 | [diff] [blame] | 104 | "torch/csrc/profiler/orchestration/python_tracer.cpp", |
Louis Feng | 5847cb5 | 2023-06-22 19:41:54 +0000 | [diff] [blame] | 105 | "torch/csrc/profiler/standalone/execution_trace_observer.cpp", |
Taylor Robie | 35fb007 | 2022-10-13 07:49:00 -0700 | [diff] [blame] | 106 | "torch/csrc/profiler/standalone/itt_observer.cpp", |
| 107 | "torch/csrc/profiler/standalone/nvtx_observer.cpp", |
Taylor Robie | b8f14b7 | 2022-10-13 07:48:58 -0700 | [diff] [blame] | 108 | "torch/csrc/profiler/stubs/base.cpp", |
Salil Desai | 8c80a46 | 2022-12-18 18:56:20 -0800 | [diff] [blame] | 109 | "torch/csrc/profiler/orchestration/vulkan.cpp", |
Digant Desai | 0fc7de3 | 2022-11-01 21:39:03 -0700 | [diff] [blame] | 110 | "torch/csrc/profiler/perf.cpp", |
Tristan Rice | 758d7de | 2021-11-18 21:44:53 -0800 | [diff] [blame] | 111 | "torch/csrc/monitor/counters.cpp", |
Jiawei Lv | b4c4a01 | 2021-12-16 11:10:28 -0800 | [diff] [blame] | 112 | "torch/csrc/monitor/events.cpp", |
Dhruv Matani | 8c852de | 2021-06-06 13:14:28 -0700 | [diff] [blame] | 113 | ] |
| 114 | |
Kimish Patel | 38c1851 | 2021-08-13 21:37:57 -0700 | [diff] [blame] | 115 | libtorch_edge_profiler_sources = libtorch_profiler_sources + [ |
| 116 | "torch/csrc/jit/mobile/profiler_edge.cpp", |
| 117 | ] |
| 118 | |
Ann Shan | 330a107 | 2020-07-27 15:36:37 -0700 | [diff] [blame] | 119 | core_trainer_sources = [ |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 120 | "torch/csrc/autograd/anomaly_mode.cpp", |
| 121 | "torch/csrc/autograd/autograd.cpp", |
soulitzer | 90a6498 | 2021-08-27 14:59:08 -0700 | [diff] [blame] | 122 | "torch/csrc/autograd/autograd_not_implemented_fallback.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 123 | "torch/csrc/autograd/cpp_hook.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 124 | "torch/csrc/autograd/custom_function.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 125 | "torch/csrc/autograd/engine.cpp", |
| 126 | "torch/csrc/autograd/function.cpp", |
soulitzer | d5007d8 | 2023-11-06 15:22:50 -0500 | [diff] [blame] | 127 | "torch/csrc/autograd/input_metadata.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 128 | "torch/csrc/autograd/functions/accumulate_grad.cpp", |
| 129 | "torch/csrc/autograd/functions/basic_ops.cpp", |
| 130 | "torch/csrc/autograd/functions/tensor.cpp", |
| 131 | "torch/csrc/autograd/functions/utils.cpp", |
| 132 | "torch/csrc/autograd/input_buffer.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 133 | "torch/csrc/autograd/record_function_ops.cpp", |
| 134 | "torch/csrc/autograd/saved_variable.cpp", |
| 135 | "torch/csrc/autograd/variable.cpp", |
Peter Bell | 5f45927 | 2021-10-13 15:33:27 -0700 | [diff] [blame] | 136 | "torch/csrc/autograd/utils/warnings.cpp", |
soulitzer | 7f88934 | 2022-09-15 22:46:16 +0000 | [diff] [blame] | 137 | "torch/csrc/autograd/jit_decomp_interface.cpp", |
Ann Shan | 330a107 | 2020-07-27 15:36:37 -0700 | [diff] [blame] | 138 | "torch/csrc/jit/frontend/name_mangler.cpp", |
| 139 | "torch/csrc/jit/ir/type_hashing.cpp", |
| 140 | "torch/csrc/jit/serialization/pickler.cpp", |
| 141 | "torch/csrc/jit/serialization/type_name_uniquer.cpp", |
Martin Yuan | 7a33d8b | 2020-07-15 21:13:24 -0700 | [diff] [blame] | 142 | ] |
| 143 | |
Han Qi (qihqi) | 25eb7c3 | 2022-12-08 03:48:04 +0000 | [diff] [blame] | 144 | torch_mobile_core = [ |
| 145 | # backend_debug_info.cpp provides |
| 146 | # __torch__.torch.classes.backend.BackendDebugInfo class |
| 147 | # This should not be needed eventually. |
| 148 | # TODO: Remove this dependency |
| 149 | "torch/csrc/jit/backends/backend_debug_info.cpp", |
| 150 | "torch/csrc/jit/mobile/compatibility/model_compatibility.cpp", |
| 151 | "torch/csrc/jit/mobile/function.cpp", |
| 152 | "torch/csrc/jit/mobile/import.cpp", |
| 153 | "torch/csrc/jit/mobile/flatbuffer_loader.cpp", |
| 154 | "torch/csrc/jit/mobile/interpreter.cpp", |
| 155 | "torch/csrc/jit/mobile/module.cpp", |
| 156 | "torch/csrc/jit/mobile/observer.cpp", |
| 157 | "torch/csrc/jit/mobile/parse_bytecode.cpp", |
| 158 | "torch/csrc/jit/mobile/parse_operators.cpp", |
| 159 | "torch/csrc/jit/mobile/quantization.cpp", |
| 160 | "torch/csrc/jit/mobile/upgrader_mobile.cpp", |
| 161 | "torch/csrc/jit/runtime/register_prim_ops.cpp", |
| 162 | "torch/csrc/jit/runtime/register_special_ops.cpp", |
| 163 | ] |
| 164 | |
| 165 | core_sources_full_mobile_no_backend_interface_xplat = [ |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 166 | "torch/csrc/jit/api/function_impl.cpp", |
| 167 | "torch/csrc/jit/api/module.cpp", |
| 168 | "torch/csrc/jit/api/object.cpp", |
Kimish Patel | e0fc473 | 2021-05-04 09:17:43 -0700 | [diff] [blame] | 169 | "torch/csrc/jit/backends/backend_debug_handler.cpp", |
Meghan Lele | 3e01931 | 2020-07-07 20:53:22 -0700 | [diff] [blame] | 170 | "torch/csrc/jit/backends/backend_detail.cpp", |
Nikita Shulga | f235c65 | 2021-02-17 07:15:45 -0800 | [diff] [blame] | 171 | "torch/csrc/jit/backends/backend_resolver.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 172 | "torch/csrc/jit/codegen/fuser/codegen.cpp", |
| 173 | "torch/csrc/jit/codegen/fuser/compiler.cpp", |
| 174 | "torch/csrc/jit/codegen/fuser/executor.cpp", |
| 175 | "torch/csrc/jit/codegen/fuser/fallback.cpp", |
| 176 | "torch/csrc/jit/codegen/fuser/interface.cpp", |
| 177 | "torch/csrc/jit/codegen/fuser/kernel_cache.cpp", |
| 178 | "torch/csrc/jit/frontend/builtin_functions.cpp", |
Mike Ruberry | 30fabd9 | 2020-04-16 04:54:26 -0700 | [diff] [blame] | 179 | "torch/csrc/jit/frontend/versioned_symbols.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 180 | "torch/csrc/jit/frontend/canonicalize_modified_loop.cpp", |
| 181 | "torch/csrc/jit/frontend/convert_to_ssa.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 182 | "torch/csrc/jit/frontend/exit_transforms.cpp", |
| 183 | "torch/csrc/jit/frontend/inline_loop_condition.cpp", |
| 184 | "torch/csrc/jit/frontend/ir_emitter.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 185 | "torch/csrc/jit/frontend/parser.cpp", |
| 186 | "torch/csrc/jit/frontend/schema_matching.cpp", |
| 187 | "torch/csrc/jit/frontend/script_type_parser.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 188 | "torch/csrc/jit/frontend/sugared_value.cpp", |
| 189 | "torch/csrc/jit/frontend/tracer.cpp", |
| 190 | "torch/csrc/jit/ir/alias_analysis.cpp", |
| 191 | "torch/csrc/jit/ir/attributes.cpp", |
| 192 | "torch/csrc/jit/ir/constants.cpp", |
| 193 | "torch/csrc/jit/ir/ir.cpp", |
| 194 | "torch/csrc/jit/ir/irparser.cpp", |
| 195 | "torch/csrc/jit/ir/node_hashing.cpp", |
| 196 | "torch/csrc/jit/ir/scope.cpp", |
| 197 | "torch/csrc/jit/ir/subgraph_matcher.cpp", |
Maxwell Nuyens | 0d0ebcd | 2023-02-10 17:12:52 +0000 | [diff] [blame] | 198 | "torch/csrc/jit/ir/graph_utils.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 199 | "torch/csrc/jit/jit_log.cpp", |
Tugsbayasgalan Manlaibaatar | 559e2d8 | 2021-01-11 12:22:54 -0800 | [diff] [blame] | 200 | "torch/csrc/jit/jit_opt_limit.cpp", |
Priya Ramani | 206646d | 2021-09-15 19:12:47 -0700 | [diff] [blame] | 201 | "torch/csrc/jit/mobile/nnc/aot_compiler.cpp", |
Jiakai Liu | 5824a86 | 2021-06-19 06:09:31 -0700 | [diff] [blame] | 202 | "torch/csrc/jit/mobile/nnc/backend.cpp", |
| 203 | "torch/csrc/jit/mobile/nnc/context.cpp", |
| 204 | "torch/csrc/jit/mobile/nnc/registry.cpp", |
Tugsbayasgalan (Tugsuu) Manlaibaatar | df3cbcf | 2021-12-24 12:21:40 -0800 | [diff] [blame] | 205 | "torch/csrc/jit/operator_upgraders/utils.cpp", |
Tugsbayasgalan (Tugsuu) Manlaibaatar | b0fdca8 | 2022-01-05 23:55:49 -0800 | [diff] [blame] | 206 | "torch/csrc/jit/operator_upgraders/upgraders.cpp", |
Tugsbayasgalan (Tugsuu) Manlaibaatar | c9bd1c6 | 2022-01-21 16:17:03 -0800 | [diff] [blame] | 207 | "torch/csrc/jit/operator_upgraders/upgraders_entry.cpp", |
Mike Iovine | d1c5f9e | 2022-02-17 10:18:33 -0800 | [diff] [blame] | 208 | "torch/csrc/jit/passes/add_if_then_else.cpp", |
Yanan Cao | d150d3e | 2020-10-02 14:39:14 -0700 | [diff] [blame] | 209 | "torch/csrc/jit/passes/annotate_warns.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 210 | "torch/csrc/jit/passes/bailout_graph.cpp", |
Elias Ellison | 0d7be81 | 2022-04-25 16:08:57 +0000 | [diff] [blame] | 211 | "torch/csrc/jit/passes/check_strict_fusion.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 212 | "torch/csrc/jit/passes/batch_mm.cpp", |
| 213 | "torch/csrc/jit/passes/canonicalize.cpp", |
Elias Ellison | 5183e3a | 2020-05-21 21:40:03 -0700 | [diff] [blame] | 214 | "torch/csrc/jit/passes/canonicalize_graph_fuser_ops.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 215 | "torch/csrc/jit/passes/clear_profiling.cpp", |
| 216 | "torch/csrc/jit/passes/clear_undefinedness.cpp", |
| 217 | "torch/csrc/jit/passes/common_subexpression_elimination.cpp", |
Raghavan Raman | 259d19a | 2021-05-09 22:05:41 -0700 | [diff] [blame] | 218 | "torch/csrc/jit/passes/concat_opt.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 219 | "torch/csrc/jit/passes/constant_pooling.cpp", |
| 220 | "torch/csrc/jit/passes/constant_propagation.cpp", |
Bin Bao | add291c | 2021-06-03 06:41:43 -0700 | [diff] [blame] | 221 | "torch/csrc/jit/passes/restore_mutation.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 222 | "torch/csrc/jit/passes/create_autodiff_subgraphs.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 223 | "torch/csrc/jit/passes/dead_code_elimination.cpp", |
Mike Iovine | 5bc8927 | 2021-11-08 08:39:58 -0800 | [diff] [blame] | 224 | "torch/csrc/jit/passes/eliminate_no_ops.cpp", |
Elias Ellison | 5da8a7b | 2020-08-31 11:45:27 -0700 | [diff] [blame] | 225 | "torch/csrc/jit/passes/remove_redundant_profiles.cpp", |
Mikhail Zolotukhin | 85c056a | 2021-03-30 13:55:06 -0700 | [diff] [blame] | 226 | "torch/csrc/jit/passes/remove_exceptions.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 227 | "torch/csrc/jit/passes/decompose_ops.cpp", |
John Clow | ec8a71f | 2021-11-04 18:57:19 -0700 | [diff] [blame] | 228 | "torch/csrc/jit/passes/dtype_analysis.cpp", |
John Clow | ade83ed | 2022-01-13 13:55:26 -0800 | [diff] [blame] | 229 | "torch/csrc/jit/passes/device_type_analysis.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 230 | "torch/csrc/jit/passes/erase_number_types.cpp", |
| 231 | "torch/csrc/jit/passes/fixup_trace_scope_blocks.cpp", |
| 232 | "torch/csrc/jit/passes/freeze_module.cpp", |
| 233 | "torch/csrc/jit/passes/fuse_linear.cpp", |
Kimish Patel | c5dcf05 | 2020-07-09 16:20:22 -0700 | [diff] [blame] | 234 | "torch/csrc/jit/passes/fuse_relu.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 235 | "torch/csrc/jit/passes/graph_fuser.cpp", |
| 236 | "torch/csrc/jit/passes/graph_rewrite_helper.cpp", |
| 237 | "torch/csrc/jit/passes/guard_elimination.cpp", |
Vasiliy Kuznetsov | 79b8328 | 2020-08-08 15:49:38 -0700 | [diff] [blame] | 238 | "torch/csrc/jit/passes/hoist_conv_packed_params.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 239 | "torch/csrc/jit/passes/inline_autodiff_subgraphs.cpp", |
| 240 | "torch/csrc/jit/passes/inline_forked_closures.cpp", |
Supriya Rao | 199c73b | 2020-09-05 12:03:12 -0700 | [diff] [blame] | 241 | "torch/csrc/jit/passes/inline_fork_wait.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 242 | "torch/csrc/jit/passes/inliner.cpp", |
| 243 | "torch/csrc/jit/passes/inplace_check.cpp", |
| 244 | "torch/csrc/jit/passes/insert_guards.cpp", |
| 245 | "torch/csrc/jit/passes/lift_closures.cpp", |
| 246 | "torch/csrc/jit/passes/liveness.cpp", |
| 247 | "torch/csrc/jit/passes/loop_unrolling.cpp", |
| 248 | "torch/csrc/jit/passes/lower_grad_of.cpp", |
| 249 | "torch/csrc/jit/passes/lower_tuples.cpp", |
Elias Ellison | f90dc74 | 2020-05-21 21:40:03 -0700 | [diff] [blame] | 250 | "torch/csrc/jit/passes/normalize_ops.cpp", |
Mike Iovine | 000e3a08 | 2021-08-13 10:18:03 -0700 | [diff] [blame] | 251 | "torch/csrc/jit/passes/peephole_dict_idioms.cpp", |
Elias Ellison | f5b3125 | 2020-05-06 15:49:37 -0700 | [diff] [blame] | 252 | "torch/csrc/jit/passes/peephole_list_idioms.cpp", |
Elias Ellison | 0d9f1c1 | 2021-05-21 08:48:13 -0700 | [diff] [blame] | 253 | "torch/csrc/jit/passes/value_refinement_utils.cpp", |
Elias Ellison | 30aeed7 | 2021-01-12 11:35:08 -0800 | [diff] [blame] | 254 | "torch/csrc/jit/passes/peephole_alias_sensitive.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 255 | "torch/csrc/jit/passes/pass_manager.cpp", |
| 256 | "torch/csrc/jit/passes/peephole.cpp", |
Elias Ellison | 391603d | 2021-05-21 08:48:13 -0700 | [diff] [blame] | 257 | "torch/csrc/jit/passes/peephole_non_tensor.cpp", |
Elias Ellison | c516f84 | 2020-04-28 23:18:29 -0700 | [diff] [blame] | 258 | "torch/csrc/jit/passes/create_functional_graphs.cpp", |
Henry Tu | f6eb811 | 2022-05-12 00:48:39 +0000 | [diff] [blame] | 259 | "torch/csrc/jit/passes/refine_tuple_types.cpp", |
Elias Ellison | 3f32332 | 2020-07-08 14:31:15 -0700 | [diff] [blame] | 260 | "torch/csrc/jit/passes/remove_mutation.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 261 | "torch/csrc/jit/passes/prepack_folding.cpp", |
Jerry Zhang | 0ed7fc5 | 2020-05-08 12:21:40 -0700 | [diff] [blame] | 262 | "torch/csrc/jit/passes/fold_conv_bn.cpp", |
min-jean-cho | 6d2b0cb | 2022-12-21 08:00:32 +0000 | [diff] [blame] | 263 | "torch/csrc/jit/passes/fold_linear_bn.cpp", |
Vasiliy Kuznetsov | bf896a2 | 2022-03-03 07:17:57 -0800 | [diff] [blame] | 264 | "torch/csrc/jit/passes/dbr_quantization/remove_redundant_aliases.cpp", |
John Clow | 3bad540 | 2021-10-08 10:54:16 -0700 | [diff] [blame] | 265 | "torch/csrc/jit/passes/frozen_concat_linear.cpp", |
Bin Bao | c1a4422 | 2021-04-30 15:26:13 -0700 | [diff] [blame] | 266 | "torch/csrc/jit/passes/frozen_conv_add_relu_fusion.cpp", |
Elias Ellison | 035229c | 2021-01-12 11:35:08 -0800 | [diff] [blame] | 267 | "torch/csrc/jit/passes/frozen_conv_folding.cpp", |
min-jean-cho | 6d2b0cb | 2022-12-21 08:00:32 +0000 | [diff] [blame] | 268 | "torch/csrc/jit/passes/frozen_linear_folding.cpp", |
John Clow | 6cdea82 | 2021-10-05 20:07:01 -0700 | [diff] [blame] | 269 | "torch/csrc/jit/passes/frozen_linear_transpose.cpp", |
Elias Ellison | bfae378 | 2021-03-01 21:14:16 -0800 | [diff] [blame] | 270 | "torch/csrc/jit/passes/frozen_ops_to_mkldnn.cpp", |
Elias Ellison | a389b30 | 2021-01-12 11:35:08 -0800 | [diff] [blame] | 271 | "torch/csrc/jit/passes/frozen_graph_optimizations.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 272 | "torch/csrc/jit/passes/remove_expands.cpp", |
Kimish Patel | f954dd7 | 2020-05-12 14:36:25 -0700 | [diff] [blame] | 273 | "torch/csrc/jit/passes/remove_dropout.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 274 | "torch/csrc/jit/passes/requires_grad_analysis.cpp", |
| 275 | "torch/csrc/jit/passes/shape_analysis.cpp", |
Elias Ellison | 5313baf | 2021-05-21 08:48:13 -0700 | [diff] [blame] | 276 | "torch/csrc/jit/passes/integer_value_refinement.cpp", |
Tugsbayasgalan (Tugsuu) Manlaibaatar | 4ae71c8 | 2021-12-25 13:01:35 -0800 | [diff] [blame] | 277 | "torch/csrc/jit/passes/replacement_of_old_operators.cpp", |
Elias Ellison | f39471a | 2021-05-21 08:48:13 -0700 | [diff] [blame] | 278 | "torch/csrc/jit/passes/symbolic_shape_analysis.cpp", |
Nikolay Korovaiko | ce842f4 | 2022-04-13 00:24:52 -0700 | [diff] [blame] | 279 | "torch/csrc/jit/passes/symbolic_shape_cache.cpp", |
Elias Ellison | fc82ad1 | 2021-10-28 17:07:44 -0700 | [diff] [blame] | 280 | "torch/csrc/jit/passes/symbolic_shape_runtime_fusion.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 281 | "torch/csrc/jit/passes/specialize_autogradzero.cpp", |
Elias Ellison | 1c0faa7 | 2020-08-31 11:45:27 -0700 | [diff] [blame] | 282 | "torch/csrc/jit/passes/update_differentiable_graph_requires_grad.cpp", |
Mike Iovine | fc6dd0b | 2021-08-23 17:26:27 -0700 | [diff] [blame] | 283 | "torch/csrc/jit/passes/variadic_ops.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 284 | "torch/csrc/jit/passes/subgraph_rewrite.cpp", |
| 285 | "torch/csrc/jit/passes/tensorexpr_fuser.cpp", |
| 286 | "torch/csrc/jit/passes/utils/memory_dag.cpp", |
| 287 | "torch/csrc/jit/passes/utils/subgraph_utils.cpp", |
John Clow | 71d1d16 | 2021-09-24 13:48:19 -0700 | [diff] [blame] | 288 | "torch/csrc/jit/passes/utils/optimization_utils.cpp", |
John Clow | 77db720 | 2021-11-18 16:25:21 -0800 | [diff] [blame] | 289 | "torch/csrc/jit/passes/utils/op_registry.cpp", |
chunyuan-w | 693a8dd | 2022-08-10 21:46:51 +0000 | [diff] [blame] | 290 | "torch/csrc/jit/passes/mkldnn_rewrite.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 291 | "torch/csrc/jit/passes/xnnpack_rewrite.cpp", |
Ivan Kobzarev | 3852215 | 2020-06-20 14:08:52 -0700 | [diff] [blame] | 292 | "torch/csrc/jit/passes/vulkan_rewrite.cpp", |
Tao Xu | a277c09 | 2020-10-13 01:44:36 -0700 | [diff] [blame] | 293 | "torch/csrc/jit/passes/metal_rewrite.cpp", |
Jerry Zhang | 0ed7fc5 | 2020-05-08 12:21:40 -0700 | [diff] [blame] | 294 | "torch/csrc/jit/passes/quantization/helper.cpp", |
Jerry Zhang | fd7e09a | 2020-06-17 23:33:01 -0700 | [diff] [blame] | 295 | "torch/csrc/jit/passes/quantization/quantization_type.cpp", |
Jerry Zhang | 0ed7fc5 | 2020-05-08 12:21:40 -0700 | [diff] [blame] | 296 | "torch/csrc/jit/passes/quantization/insert_observers.cpp", |
| 297 | "torch/csrc/jit/passes/quantization/insert_quant_dequant.cpp", |
| 298 | "torch/csrc/jit/passes/quantization/dedup_module_uses.cpp", |
| 299 | "torch/csrc/jit/passes/quantization/finalize.cpp", |
Kimish Patel | bb12e4d | 2020-05-27 14:08:24 -0700 | [diff] [blame] | 300 | "torch/csrc/jit/passes/quantization/fusion_passes.cpp", |
Kimish Patel | 5c7e801 | 2022-08-27 16:06:15 -0700 | [diff] [blame] | 301 | "torch/csrc/jit/passes/quantization/register_packed_params.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 302 | "torch/csrc/jit/python/update_graph_executor_opt.cpp", |
Shuming Hu | b45880c | 2023-03-23 01:19:08 +0000 | [diff] [blame] | 303 | "torch/csrc/jit/python/utf8_decoding_ignore.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 304 | "torch/csrc/jit/runtime/argument_spec.cpp", |
| 305 | "torch/csrc/jit/runtime/autodiff.cpp", |
| 306 | "torch/csrc/jit/runtime/graph_executor.cpp", |
Zhengxu Chen | 8b38458 | 2021-05-06 16:42:41 -0700 | [diff] [blame] | 307 | "torch/csrc/jit/runtime/interpreter/frame.cpp", |
| 308 | "torch/csrc/jit/runtime/interpreter/preprocess_graph.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 309 | "torch/csrc/jit/runtime/interpreter.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 310 | "torch/csrc/jit/runtime/logging.cpp", |
Elias Ellison | 0ecf1ad | 2022-03-29 11:32:31 -0700 | [diff] [blame] | 311 | "torch/csrc/jit/runtime/simple_graph_executor_impl.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 312 | "torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp", |
| 313 | "torch/csrc/jit/runtime/profiling_record.cpp", |
Zhengxu Chen | 2b0ec9c | 2021-05-24 18:22:01 -0700 | [diff] [blame] | 314 | "torch/csrc/jit/runtime/script_profile.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 315 | "torch/csrc/jit/runtime/symbolic_script.cpp", |
eellison | f66fbb1 | 2021-05-27 17:52:46 -0700 | [diff] [blame] | 316 | "torch/csrc/jit/runtime/symbolic_shape_registry.cpp", |
Elias Ellison | aacdf29 | 2022-03-29 11:32:31 -0700 | [diff] [blame] | 317 | "torch/csrc/jit/runtime/decomposition_registry.cpp", |
| 318 | "torch/csrc/jit/runtime/decomposition_registry_util.cpp", |
Elias Ellison | f65eb09 | 2022-04-19 20:59:42 +0000 | [diff] [blame] | 319 | "torch/csrc/jit/runtime/serialized_shape_function_registry.cpp", |
Elias Ellison | 19b8729 | 2021-12-01 00:42:05 -0800 | [diff] [blame] | 320 | "torch/csrc/jit/runtime/symbolic_shape_registry_util.cpp", |
Nikolay Korovaiko | a7ebf76 | 2021-10-24 18:03:00 -0700 | [diff] [blame] | 321 | "torch/csrc/jit/runtime/jit_trace.cpp", |
Kimish Patel | e0fc473 | 2021-05-04 09:17:43 -0700 | [diff] [blame] | 322 | "torch/csrc/jit/serialization/callstack_debug_info_serialization.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 323 | "torch/csrc/jit/serialization/import.cpp", |
| 324 | "torch/csrc/jit/serialization/import_export_helpers.cpp", |
| 325 | "torch/csrc/jit/serialization/import_source.cpp", |
| 326 | "torch/csrc/jit/serialization/pickle.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 327 | "torch/csrc/jit/serialization/python_print.cpp", |
| 328 | "torch/csrc/jit/serialization/source_range_serialization.cpp", |
Mikhail Zolotukhin | c639513 | 2021-02-03 10:20:24 -0800 | [diff] [blame] | 329 | "torch/csrc/jit/tensorexpr/block_codegen.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 330 | "torch/csrc/jit/tensorexpr/bounds_inference.cpp", |
Nick Gibson | 0edc6a3 | 2020-11-04 19:50:25 -0800 | [diff] [blame] | 331 | "torch/csrc/jit/tensorexpr/bounds_overlap.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 332 | "torch/csrc/jit/tensorexpr/codegen.cpp", |
Cheng Chang | 109bc10 | 2021-02-01 13:04:20 -0800 | [diff] [blame] | 333 | "torch/csrc/jit/tensorexpr/cpp_codegen.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 334 | "torch/csrc/jit/tensorexpr/eval.cpp", |
| 335 | "torch/csrc/jit/tensorexpr/expr.cpp", |
Ivan Kobzarev | 519e226 | 2022-03-09 13:13:47 -0800 | [diff] [blame] | 336 | "torch/csrc/jit/tensorexpr/external_functions_core.cpp", |
Mikhail Zolotukhin | c639513 | 2021-02-03 10:20:24 -0800 | [diff] [blame] | 337 | "torch/csrc/jit/tensorexpr/external_functions_registry.cpp", |
Raghavan Raman | d0c4ace | 2021-06-18 14:29:14 -0700 | [diff] [blame] | 338 | "torch/csrc/jit/tensorexpr/graph_opt.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 339 | "torch/csrc/jit/tensorexpr/hash_provider.cpp", |
Bert Maher | 0a9764e | 2021-01-28 22:29:23 -0800 | [diff] [blame] | 340 | "torch/csrc/jit/tensorexpr/intrinsic_symbols.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 341 | "torch/csrc/jit/tensorexpr/ir.cpp", |
Raghavan Raman | e54ee9b | 2021-08-09 09:11:13 -0700 | [diff] [blame] | 342 | "torch/csrc/jit/tensorexpr/ir_cloner.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 343 | "torch/csrc/jit/tensorexpr/ir_mutator.cpp", |
| 344 | "torch/csrc/jit/tensorexpr/ir_printer.cpp", |
| 345 | "torch/csrc/jit/tensorexpr/ir_simplifier.cpp", |
Mikhail Zolotukhin | e22da0a | 2021-03-01 20:35:17 -0800 | [diff] [blame] | 346 | "torch/csrc/jit/tensorexpr/ir_verifier.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 347 | "torch/csrc/jit/tensorexpr/ir_visitor.cpp", |
| 348 | "torch/csrc/jit/tensorexpr/kernel.cpp", |
| 349 | "torch/csrc/jit/tensorexpr/llvm_codegen.cpp", |
| 350 | "torch/csrc/jit/tensorexpr/llvm_jit.cpp", |
| 351 | "torch/csrc/jit/tensorexpr/loopnest.cpp", |
Animesh Jain | 6896b2d | 2022-01-06 16:20:13 -0800 | [diff] [blame] | 352 | "torch/csrc/jit/tensorexpr/loopnest_randomization.cpp", |
Mikhail Zolotukhin | eee9ad0 | 2021-09-30 22:51:23 -0700 | [diff] [blame] | 353 | "torch/csrc/jit/tensorexpr/lowerings.cpp", |
Mikhail Zolotukhin | c639513 | 2021-02-03 10:20:24 -0800 | [diff] [blame] | 354 | "torch/csrc/jit/tensorexpr/mem_dependency_checker.cpp", |
Bert Maher | 90f8485 | 2021-04-08 21:54:52 -0700 | [diff] [blame] | 355 | "torch/csrc/jit/tensorexpr/operators/conv2d.cpp", |
Bert Maher | bda4063 | 2021-06-16 05:07:43 -0700 | [diff] [blame] | 356 | "torch/csrc/jit/tensorexpr/operators/matmul.cpp", |
Mikhail Zolotukhin | 015e007 | 2021-09-30 22:51:23 -0700 | [diff] [blame] | 357 | "torch/csrc/jit/tensorexpr/operators/misc.cpp", |
Bert Maher | 842a831 | 2021-06-16 05:07:43 -0700 | [diff] [blame] | 358 | "torch/csrc/jit/tensorexpr/operators/norm.cpp", |
Mikhail Zolotukhin | 015e007 | 2021-09-30 22:51:23 -0700 | [diff] [blame] | 359 | "torch/csrc/jit/tensorexpr/operators/pointwise.cpp", |
Ivan Kobzarev | 7fbcf79 | 2021-10-31 22:48:04 -0700 | [diff] [blame] | 360 | "torch/csrc/jit/tensorexpr/operators/quantization.cpp", |
Bert Maher | bda4063 | 2021-06-16 05:07:43 -0700 | [diff] [blame] | 361 | "torch/csrc/jit/tensorexpr/operators/reduction.cpp", |
| 362 | "torch/csrc/jit/tensorexpr/operators/softmax.cpp", |
Nick Gibson | eab8093 | 2020-11-12 20:15:47 -0800 | [diff] [blame] | 363 | "torch/csrc/jit/tensorexpr/reduction.cpp", |
Nick Gibson | aabdef5 | 2020-08-11 11:15:38 -0700 | [diff] [blame] | 364 | "torch/csrc/jit/tensorexpr/registerizer.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 365 | "torch/csrc/jit/tensorexpr/tensor.cpp", |
| 366 | "torch/csrc/jit/tensorexpr/types.cpp", |
| 367 | "torch/csrc/jit/tensorexpr/unique_name_manager.cpp", |
| 368 | "torch/csrc/jit/testing/file_check.cpp", |
PyTorch MergeBot | 2c313e7 | 2023-06-12 23:52:10 +0000 | [diff] [blame] | 369 | "torch/csrc/profiler/unwind/unwind.cpp", |
zdevito | 3fe3439 | 2023-10-03 16:42:11 -0700 | [diff] [blame] | 370 | "torch/csrc/profiler/unwind/unwind_fb.cpp", |
PyTorch MergeBot | 2c313e7 | 2023-06-12 23:52:10 +0000 | [diff] [blame] | 371 | "torch/csrc/profiler/combined_traceback.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 372 | "torch/csrc/jit/testing/hooks_for_testing.cpp", |
David Berard | 53b4f6c | 2023-03-02 07:38:51 -0800 | [diff] [blame] | 373 | "torch/csrc/utils/cpp_stacktraces.cpp", |
goldenxuett | b4e3429 | 2022-07-08 11:31:31 -0700 | [diff] [blame] | 374 | "torch/csrc/utils/schema_info.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 375 | "torch/csrc/utils/tensor_flatten.cpp", |
| 376 | "torch/csrc/utils/variadic.cpp", |
Kimish Patel | 4680016 | 2021-09-01 12:38:39 -0700 | [diff] [blame] | 377 | ] |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 378 | |
Han Qi (qihqi) | 25eb7c3 | 2022-12-08 03:48:04 +0000 | [diff] [blame] | 379 | core_sources_full_mobile_no_backend_interface = core_sources_full_mobile_no_backend_interface_xplat + [ |
| 380 | # backend_debug_info.cpp provides |
| 381 | # __torch__.torch.classes.backend.BackendDebugInfo class |
| 382 | # This should not be needed eventually. |
| 383 | # TODO: Remove this dependency |
| 384 | "torch/csrc/jit/backends/backend_debug_info.cpp", |
| 385 | "torch/csrc/jit/mobile/compatibility/model_compatibility.cpp", |
| 386 | "torch/csrc/jit/mobile/function.cpp", |
| 387 | "torch/csrc/jit/mobile/import.cpp", |
| 388 | "torch/csrc/jit/mobile/flatbuffer_loader.cpp", |
| 389 | "torch/csrc/jit/mobile/interpreter.cpp", |
| 390 | "torch/csrc/jit/mobile/module.cpp", |
| 391 | "torch/csrc/jit/mobile/observer.cpp", |
| 392 | "torch/csrc/jit/mobile/parse_bytecode.cpp", |
| 393 | "torch/csrc/jit/mobile/parse_operators.cpp", |
| 394 | "torch/csrc/jit/mobile/quantization.cpp", |
| 395 | "torch/csrc/jit/mobile/upgrader_mobile.cpp", |
| 396 | ] |
| 397 | |
| 398 | |
Amy He | 63554cf | 2021-08-17 10:31:02 -0700 | [diff] [blame] | 399 | core_sources_full_mobile = core_sources_full_mobile_no_backend_interface + [ |
| 400 | "torch/csrc/jit/backends/backend_debug_info.cpp", |
| 401 | "torch/csrc/jit/backends/backend_interface.cpp", |
| 402 | ] |
| 403 | |
Martin Yuan | 6615eda | 2020-11-20 22:24:46 -0800 | [diff] [blame] | 404 | core_sources_full = core_sources_full_mobile + [ |
Bram Wasti | f4226b5 | 2020-12-10 14:01:36 -0800 | [diff] [blame] | 405 | "torch/csrc/jit/runtime/static/fusion.cpp", |
Don Jang | fe7e1bd | 2022-02-28 00:29:16 -0800 | [diff] [blame] | 406 | "torch/csrc/jit/runtime/static/generated_ops.cpp", |
Martin Yuan | 6615eda | 2020-11-20 22:24:46 -0800 | [diff] [blame] | 407 | "torch/csrc/jit/runtime/static/impl.cpp", |
Don Jang | ae00075 | 2021-09-17 13:20:33 -0700 | [diff] [blame] | 408 | "torch/csrc/jit/runtime/static/memory_planner.cpp", |
Hao Lu | a07b081 | 2021-07-14 02:12:20 -0700 | [diff] [blame] | 409 | "torch/csrc/jit/runtime/static/native_ops.cpp", |
Martin Yuan | 6615eda | 2020-11-20 22:24:46 -0800 | [diff] [blame] | 410 | "torch/csrc/jit/runtime/static/ops.cpp", |
Ansha Yu | 07978bd | 2020-12-08 05:52:48 -0800 | [diff] [blame] | 411 | "torch/csrc/jit/runtime/static/passes.cpp", |
Raghavan Raman | a2399a7 | 2021-08-25 11:12:57 -0700 | [diff] [blame] | 412 | "torch/csrc/jit/runtime/static/te_wrapper.cpp", |
Mikhail Zolotukhin | c639513 | 2021-02-03 10:20:24 -0800 | [diff] [blame] | 413 | "torch/csrc/jit/tensorexpr/external_functions.cpp", |
Horace He | 31607ad | 2021-05-13 19:55:39 -0700 | [diff] [blame] | 414 | "torch/csrc/jit/tensorexpr/external_functions_codegen.cpp", |
Martin Yuan | 6615eda | 2020-11-20 22:24:46 -0800 | [diff] [blame] | 415 | ] |
| 416 | |
Will Constable | a8c0b36 | 2021-10-07 10:03:02 -0700 | [diff] [blame] | 417 | lazy_tensor_core_sources = [ |
Jiewen Tan | 6011c35 | 2021-11-10 07:02:44 -0800 | [diff] [blame] | 418 | "torch/csrc/lazy/backend/backend_device.cpp", |
Jiewen Tan | e6c435b | 2021-12-01 12:13:59 -0800 | [diff] [blame] | 419 | "torch/csrc/lazy/backend/backend_interface.cpp", |
Will Constable | d6e6064 | 2021-11-10 12:51:43 -0800 | [diff] [blame] | 420 | "torch/csrc/lazy/backend/lowering_context.cpp", |
Will Constable | d05c1ec | 2021-10-18 19:08:09 -0700 | [diff] [blame] | 421 | "torch/csrc/lazy/core/config.cpp", |
Will Constable | 328cfd5 | 2022-02-11 09:54:34 -0800 | [diff] [blame] | 422 | "torch/csrc/lazy/core/debug_util.cpp", |
Will Constable | a8c0b36 | 2021-10-07 10:03:02 -0700 | [diff] [blame] | 423 | "torch/csrc/lazy/core/hash.cpp", |
Bin Bao | 0bbe21b | 2021-12-04 08:38:39 -0800 | [diff] [blame] | 424 | "torch/csrc/lazy/core/helpers.cpp", |
Will Constable | d05c1ec | 2021-10-18 19:08:09 -0700 | [diff] [blame] | 425 | "torch/csrc/lazy/core/ir.cpp", |
Bin Bao | 0bbe21b | 2021-12-04 08:38:39 -0800 | [diff] [blame] | 426 | "torch/csrc/lazy/core/ir_dump_util.cpp", |
Will Constable | d05c1ec | 2021-10-18 19:08:09 -0700 | [diff] [blame] | 427 | "torch/csrc/lazy/core/ir_metadata.cpp", |
Bin Bao | 2366948 | 2021-10-28 08:15:33 -0700 | [diff] [blame] | 428 | "torch/csrc/lazy/core/ir_util.cpp", |
Bin Bao | fe7b644 | 2021-12-16 05:41:42 -0800 | [diff] [blame] | 429 | "torch/csrc/lazy/core/lazy_graph_executor.cpp", |
Bin Bao | e6a4988 | 2021-12-16 05:41:42 -0800 | [diff] [blame] | 430 | "torch/csrc/lazy/core/metrics.cpp", |
| 431 | "torch/csrc/lazy/core/multi_wait.cpp", |
Antonio Kim | f3f327e | 2022-04-28 02:07:02 +0000 | [diff] [blame] | 432 | "torch/csrc/lazy/core/ops/arithmetic_ir_ops.cpp", |
| 433 | "torch/csrc/lazy/core/ops/utils.cpp", |
Bin Bao | a473417 | 2021-11-09 11:59:01 -0800 | [diff] [blame] | 434 | "torch/csrc/lazy/core/permutation_util.cpp", |
Jiewen Tan | 8bed46e | 2021-11-04 14:10:26 -0700 | [diff] [blame] | 435 | "torch/csrc/lazy/core/shape.cpp", |
Will Constable | d56d530 | 2022-02-23 23:01:46 -0800 | [diff] [blame] | 436 | "torch/csrc/lazy/core/shape_inference.cpp", |
Bin Bao | fe7b644 | 2021-12-16 05:41:42 -0800 | [diff] [blame] | 437 | "torch/csrc/lazy/core/tensor.cpp", |
Jiewen Tan | e02d836 | 2021-12-22 03:20:35 -0800 | [diff] [blame] | 438 | "torch/csrc/lazy/core/tensor_impl.cpp", |
Bin Bao | 8a975c0 | 2021-12-07 16:45:11 -0800 | [diff] [blame] | 439 | "torch/csrc/lazy/core/tensor_util.cpp", |
Bin Bao | e6a4988 | 2021-12-16 05:41:42 -0800 | [diff] [blame] | 440 | "torch/csrc/lazy/core/thread_pool.cpp", |
Bin Bao | f05710d | 2022-05-04 17:46:00 +0000 | [diff] [blame] | 441 | "torch/csrc/lazy/core/trie.cpp", |
Will Constable | a8c0b36 | 2021-10-07 10:03:02 -0700 | [diff] [blame] | 442 | ] |
| 443 | |
Will Constable | 3547f20 | 2022-03-22 16:06:04 -0700 | [diff] [blame] | 444 | # We can't build all of the ts backend under certain build configurations, e.g. mobile, |
| 445 | # since it depends on things like autograd, meta functions, which may be disabled |
| 446 | lazy_tensor_ts_sources = [ |
Antonio Kim | f3f327e | 2022-04-28 02:07:02 +0000 | [diff] [blame] | 447 | "torch/csrc/lazy/ts_backend/dynamic_ir.cpp", |
Antonio Kim | 02c4d877 | 2022-05-24 19:29:23 +0000 | [diff] [blame] | 448 | "torch/csrc/lazy/ts_backend/config.cpp", |
Antonio Kim | f3f327e | 2022-04-28 02:07:02 +0000 | [diff] [blame] | 449 | "torch/csrc/lazy/ts_backend/ops/device_data.cpp", |
Antonio Kim | f3f327e | 2022-04-28 02:07:02 +0000 | [diff] [blame] | 450 | "torch/csrc/lazy/ts_backend/ops/generic.cpp", |
Antonio Kim | f3f327e | 2022-04-28 02:07:02 +0000 | [diff] [blame] | 451 | "torch/csrc/lazy/ts_backend/tensor_aten_ops.cpp", |
Will Constable | 3547f20 | 2022-03-22 16:06:04 -0700 | [diff] [blame] | 452 | "torch/csrc/lazy/ts_backend/ts_autograd_functions.cpp", |
| 453 | "torch/csrc/lazy/ts_backend/ts_backend_impl.cpp", |
Antonio Kim | f3f327e | 2022-04-28 02:07:02 +0000 | [diff] [blame] | 454 | "torch/csrc/lazy/ts_backend/ts_eager_fallback.cpp", |
Will Constable | 3547f20 | 2022-03-22 16:06:04 -0700 | [diff] [blame] | 455 | "torch/csrc/lazy/ts_backend/ts_lowering_context.cpp", |
| 456 | "torch/csrc/lazy/ts_backend/ts_native_functions.cpp", |
Antonio Kim | 02c4d877 | 2022-05-24 19:29:23 +0000 | [diff] [blame] | 457 | "torch/csrc/lazy/ts_backend/ts_node.cpp", |
Will Constable | 3547f20 | 2022-03-22 16:06:04 -0700 | [diff] [blame] | 458 | "torch/csrc/lazy/ts_backend/ts_node_lowering.cpp", |
Will Constable | 3547f20 | 2022-03-22 16:06:04 -0700 | [diff] [blame] | 459 | ] |
| 460 | |
Will Constable | 328cfd5 | 2022-02-11 09:54:34 -0800 | [diff] [blame] | 461 | lazy_tensor_core_python_sources = [ |
| 462 | "torch/csrc/lazy/python/init.cpp", |
| 463 | "torch/csrc/lazy/python/python_util.cpp", |
| 464 | ] |
| 465 | |
Bin Bao | 528ab47 | 2023-07-14 14:35:29 +0000 | [diff] [blame] | 466 | inductor_core_resources = [ |
Mu-Chu Lee | eddce3c | 2023-11-14 14:08:25 -0800 | [diff] [blame] | 467 | "torch/csrc/inductor/aoti_model_container_runner.cpp", |
Bin Bao | 0f646b1 | 2023-09-16 16:46:26 +0000 | [diff] [blame] | 468 | "torch/csrc/inductor/aoti_torch/shim_common.cpp", |
Bin Bao | 9c2715b | 2023-09-21 00:25:24 +0000 | [diff] [blame] | 469 | "torch/csrc/inductor/aoti_torch/tensor_converter.cpp", |
Bin Bao | 528ab47 | 2023-07-14 14:35:29 +0000 | [diff] [blame] | 470 | "torch/csrc/inductor/inductor_ops.cpp", |
| 471 | ] |
| 472 | |
Will Constable | a8c0b36 | 2021-10-07 10:03:02 -0700 | [diff] [blame] | 473 | libtorch_core_sources = sorted( |
| 474 | core_sources_common + |
Pavithran Ramachandran | c2ceba8 | 2021-11-04 15:52:03 -0700 | [diff] [blame] | 475 | torch_unpickler_common + |
Will Constable | a8c0b36 | 2021-10-07 10:03:02 -0700 | [diff] [blame] | 476 | core_sources_full + |
| 477 | core_trainer_sources + |
Bin Bao | 528ab47 | 2023-07-14 14:35:29 +0000 | [diff] [blame] | 478 | inductor_core_resources + |
Will Constable | a8c0b36 | 2021-10-07 10:03:02 -0700 | [diff] [blame] | 479 | libtorch_profiler_sources + |
| 480 | lazy_tensor_core_sources, |
| 481 | ) |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 482 | |
Luca Wehrstedt | a178043 | 2021-06-15 02:00:08 -0700 | [diff] [blame] | 483 | # These files are the only ones that are supported on Windows. |
| 484 | libtorch_distributed_base_sources = [ |
Howard Huang | 693ed8b | 2022-08-31 11:42:08 -0700 | [diff] [blame] | 485 | "torch/csrc/distributed/c10d/Backend.cpp", |
Luca Wehrstedt | a016150 | 2021-06-24 12:37:29 -0700 | [diff] [blame] | 486 | "torch/csrc/distributed/c10d/FileStore.cpp", |
Yifu Wang | ec18ef6 | 2023-10-25 10:49:20 -0700 | [diff] [blame] | 487 | "torch/csrc/distributed/c10d/Functional.cpp", |
Luca Wehrstedt | a016150 | 2021-06-24 12:37:29 -0700 | [diff] [blame] | 488 | "torch/csrc/distributed/c10d/GlooDeviceFactory.cpp", |
Yifu Wang | ec18ef6 | 2023-10-25 10:49:20 -0700 | [diff] [blame] | 489 | "torch/csrc/distributed/c10d/GroupRegistry.cpp", |
Jiewen Tan | e757cf4 | 2022-06-13 20:33:28 +0000 | [diff] [blame] | 490 | "torch/csrc/distributed/c10d/Ops.cpp", |
Luca Wehrstedt | a016150 | 2021-06-24 12:37:29 -0700 | [diff] [blame] | 491 | "torch/csrc/distributed/c10d/ParamCommsUtils.cpp", |
| 492 | "torch/csrc/distributed/c10d/PrefixStore.cpp", |
| 493 | "torch/csrc/distributed/c10d/ProcessGroup.cpp", |
| 494 | "torch/csrc/distributed/c10d/ProcessGroupGloo.cpp", |
| 495 | "torch/csrc/distributed/c10d/ProcessGroupMPI.cpp", |
| 496 | "torch/csrc/distributed/c10d/ProcessGroupWrapper.cpp", |
Luca Wehrstedt | a016150 | 2021-06-24 12:37:29 -0700 | [diff] [blame] | 497 | "torch/csrc/distributed/c10d/Store.cpp", |
| 498 | "torch/csrc/distributed/c10d/TCPStore.cpp", |
Rodrigo Kumpera | fe284b0 | 2023-07-24 06:51:51 -0700 | [diff] [blame] | 499 | "torch/csrc/distributed/c10d/TCPStoreBackend.cpp", |
Rodrigo Kumpera | 2636751 | 2023-07-26 10:23:14 -0700 | [diff] [blame] | 500 | "torch/csrc/distributed/c10d/TCPStoreLibUvBackend.cpp", |
Luca Wehrstedt | a016150 | 2021-06-24 12:37:29 -0700 | [diff] [blame] | 501 | "torch/csrc/distributed/c10d/Utils.cpp", |
Can Balioglu | 6e640a0 | 2021-11-16 20:47:57 -0800 | [diff] [blame] | 502 | "torch/csrc/distributed/c10d/comm.cpp", |
Can Balioglu | e1db2f1 | 2022-02-23 18:25:26 -0800 | [diff] [blame] | 503 | "torch/csrc/distributed/c10d/debug.cpp", |
Can Balioglu | 6e640a0 | 2021-11-16 20:47:57 -0800 | [diff] [blame] | 504 | "torch/csrc/distributed/c10d/default_comm_hooks.cpp", |
Can Balioglu | 6e640a0 | 2021-11-16 20:47:57 -0800 | [diff] [blame] | 505 | "torch/csrc/distributed/c10d/logger.cpp", |
Can Balioglu | e143f98 | 2022-02-23 18:25:26 -0800 | [diff] [blame] | 506 | "torch/csrc/distributed/c10d/logging.cpp", |
Tristan Rice | ed51b93 | 2022-05-13 23:38:31 +0000 | [diff] [blame] | 507 | "torch/csrc/distributed/c10d/quantization/quantization.cpp", |
Can Balioglu | 6e640a0 | 2021-11-16 20:47:57 -0800 | [diff] [blame] | 508 | "torch/csrc/distributed/c10d/reducer.cpp", |
| 509 | "torch/csrc/distributed/c10d/sequence_num.cpp", |
| 510 | "torch/csrc/distributed/c10d/socket.cpp", |
Howard Huang | 74ead61 | 2022-09-13 12:07:22 -0700 | [diff] [blame] | 511 | "torch/csrc/distributed/c10d/Work.cpp", |
Luca Wehrstedt | a178043 | 2021-06-15 02:00:08 -0700 | [diff] [blame] | 512 | ] |
| 513 | |
| 514 | # These files are only supported on Linux (and others) but not on Windows. |
| 515 | libtorch_distributed_extra_sources = [ |
Wanchao Liang | d493918 | 2020-06-08 19:40:00 -0700 | [diff] [blame] | 516 | "torch/csrc/distributed/autograd/autograd.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 517 | "torch/csrc/distributed/autograd/utils.cpp", |
| 518 | "torch/csrc/distributed/autograd/context/container.cpp", |
| 519 | "torch/csrc/distributed/autograd/context/context.cpp", |
| 520 | "torch/csrc/distributed/autograd/engine/dist_engine.cpp", |
| 521 | "torch/csrc/distributed/autograd/functions/recvrpc_backward.cpp", |
| 522 | "torch/csrc/distributed/autograd/functions/sendrpc_backward.cpp", |
| 523 | "torch/csrc/distributed/autograd/rpc_messages/autograd_metadata.cpp", |
| 524 | "torch/csrc/distributed/autograd/rpc_messages/propagate_gradients_req.cpp", |
| 525 | "torch/csrc/distributed/autograd/rpc_messages/propagate_gradients_resp.cpp", |
| 526 | "torch/csrc/distributed/autograd/rpc_messages/cleanup_autograd_context_req.cpp", |
| 527 | "torch/csrc/distributed/autograd/rpc_messages/cleanup_autograd_context_resp.cpp", |
| 528 | "torch/csrc/distributed/autograd/rpc_messages/rpc_with_autograd.cpp", |
Rohan Varma | 7e82382 | 2020-06-18 16:55:56 -0700 | [diff] [blame] | 529 | "torch/csrc/distributed/autograd/rpc_messages/rpc_with_profiling_req.cpp", |
| 530 | "torch/csrc/distributed/autograd/rpc_messages/rpc_with_profiling_resp.cpp", |
Pritam Damania | 781e0ed | 2020-11-07 21:02:48 -0800 | [diff] [blame] | 531 | "torch/csrc/distributed/autograd/rpc_messages/rref_backward_req.cpp", |
| 532 | "torch/csrc/distributed/autograd/rpc_messages/rref_backward_resp.cpp", |
Luca Wehrstedt | a016150 | 2021-06-24 12:37:29 -0700 | [diff] [blame] | 533 | "torch/csrc/distributed/c10d/HashStore.cpp", |
| 534 | "torch/csrc/distributed/c10d/ProcessGroupRoundRobin.cpp", |
Luca Wehrstedt | 08ce5ee | 2021-06-18 05:12:27 -0700 | [diff] [blame] | 535 | "torch/csrc/distributed/rpc/agent_utils.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 536 | "torch/csrc/distributed/rpc/message.cpp", |
Rohan Varma | 14f7e95 | 2020-06-22 10:57:24 -0700 | [diff] [blame] | 537 | "torch/csrc/distributed/rpc/profiler/remote_profiler_manager.cpp", |
Shihao Xu | 45baf0e | 2020-06-01 12:33:39 -0700 | [diff] [blame] | 538 | "torch/csrc/distributed/rpc/profiler/server_process_global_profiler.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 539 | "torch/csrc/distributed/rpc/python_call.cpp", |
| 540 | "torch/csrc/distributed/rpc/python_remote_call.cpp", |
| 541 | "torch/csrc/distributed/rpc/python_resp.cpp", |
| 542 | "torch/csrc/distributed/rpc/request_callback.cpp", |
Pritam Damania | ff6e560 | 2020-07-15 12:54:13 -0700 | [diff] [blame] | 543 | "torch/csrc/distributed/rpc/request_callback_no_python.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 544 | "torch/csrc/distributed/rpc/rpc_agent.cpp", |
| 545 | "torch/csrc/distributed/rpc/rref_context.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 546 | "torch/csrc/distributed/rpc/rref_impl.cpp", |
Pritam Damania | 781e0ed | 2020-11-07 21:02:48 -0800 | [diff] [blame] | 547 | "torch/csrc/distributed/rpc/rref_proto.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 548 | "torch/csrc/distributed/rpc/script_call.cpp", |
| 549 | "torch/csrc/distributed/rpc/script_remote_call.cpp", |
| 550 | "torch/csrc/distributed/rpc/script_resp.cpp", |
Luca Wehrstedt | 08ce5ee | 2021-06-18 05:12:27 -0700 | [diff] [blame] | 551 | "torch/csrc/distributed/rpc/tensorpipe_agent.cpp", |
| 552 | "torch/csrc/distributed/rpc/tensorpipe_utils.cpp", |
Howard Huang | e8d2916 | 2021-07-20 13:49:10 -0700 | [diff] [blame] | 553 | "torch/csrc/distributed/rpc/testing/faulty_tensorpipe_agent.cpp", |
Shihao Xu | 45baf0e | 2020-06-01 12:33:39 -0700 | [diff] [blame] | 554 | "torch/csrc/distributed/rpc/torchscript_functions.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 555 | "torch/csrc/distributed/rpc/types.cpp", |
| 556 | "torch/csrc/distributed/rpc/utils.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 557 | ] |
| 558 | |
Luca Wehrstedt | a178043 | 2021-06-15 02:00:08 -0700 | [diff] [blame] | 559 | libtorch_distributed_sources = libtorch_distributed_base_sources + libtorch_distributed_extra_sources |
| 560 | |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 561 | jit_sources_full = [ |
Christian Sarofeen | 6d24f8f | 2020-04-02 09:19:36 -0700 | [diff] [blame] | 562 | "torch/csrc/jit/codegen/cuda/interface.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 563 | "torch/csrc/jit/passes/lower_graph.cpp", |
| 564 | "torch/csrc/jit/runtime/register_c10_ops.cpp", |
Martin Yuan | 8a41fa4 | 2020-08-29 03:10:40 -0700 | [diff] [blame] | 565 | "torch/csrc/jit/runtime/register_prim_ops.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 566 | "torch/csrc/jit/runtime/register_prim_ops_fulljit.cpp", |
Martin Yuan | 8a41fa4 | 2020-08-29 03:10:40 -0700 | [diff] [blame] | 567 | "torch/csrc/jit/runtime/register_special_ops.cpp", |
Nikita Shulga | d7fc05b | 2020-04-16 19:08:16 -0700 | [diff] [blame] | 568 | "torch/csrc/jit/passes/remove_inplace_ops.cpp", |
| 569 | "torch/csrc/jit/passes/utils/check_alias_annotation.cpp", |
jjsjann123 | 1ec732b | 2021-10-27 12:09:53 -0700 | [diff] [blame] | 570 | "torch/csrc/jit/passes/autocast.cpp", |
Nikita Shulga | d7fc05b | 2020-04-16 19:08:16 -0700 | [diff] [blame] | 571 | ] |
| 572 | |
Raziel Alvarez Guevara | 2b38c09 | 2020-09-24 09:36:53 -0700 | [diff] [blame] | 573 | libtorch_core_jit_sources = sorted(jit_sources_full) |
Martin Yuan | b35cdc5 | 2020-06-28 16:37:23 -0700 | [diff] [blame] | 574 | |
Chen Lai | 3fe5895 | 2021-10-08 20:10:54 -0700 | [diff] [blame] | 575 | torch_mobile_tracer_sources = [ |
| 576 | "torch/csrc/jit/mobile/model_tracer/tracer.cpp", |
| 577 | "torch/csrc/jit/mobile/model_tracer/TensorUtils.cpp", |
Chen Lai | 76efbcc | 2021-10-15 02:17:57 -0700 | [diff] [blame] | 578 | "torch/csrc/jit/mobile/model_tracer/TracerRunner.cpp", |
Chen Lai | 3fe5895 | 2021-10-08 20:10:54 -0700 | [diff] [blame] | 579 | "torch/csrc/jit/mobile/model_tracer/MobileModelRunner.cpp", |
| 580 | "torch/csrc/jit/mobile/model_tracer/OperatorCallTracer.cpp", |
| 581 | "torch/csrc/jit/mobile/model_tracer/KernelDTypeTracer.cpp", |
Dhruv Matani | 18a31cc | 2022-09-09 08:51:12 -0700 | [diff] [blame] | 582 | "torch/csrc/jit/mobile/model_tracer/CustomClassTracer.cpp", |
| 583 | "torch/csrc/jit/mobile/model_tracer/BuildFeatureTracer.cpp", |
Chen Lai | 3fe5895 | 2021-10-08 20:10:54 -0700 | [diff] [blame] | 584 | ] |
| 585 | |
Kimish Patel | f4a9216 | 2021-05-04 09:17:43 -0700 | [diff] [blame] | 586 | libtorch_lite_eager_symbolication = [ |
| 587 | "torch/csrc/jit/frontend/source_range.cpp", |
Kimish Patel | e0fc473 | 2021-05-04 09:17:43 -0700 | [diff] [blame] | 588 | "torch/csrc/jit/ir/scope.cpp", |
Kimish Patel | f4a9216 | 2021-05-04 09:17:43 -0700 | [diff] [blame] | 589 | "torch/csrc/jit/mobile/debug_info.cpp", |
Kimish Patel | e0fc473 | 2021-05-04 09:17:43 -0700 | [diff] [blame] | 590 | "torch/csrc/jit/serialization/callstack_debug_info_serialization.cpp", |
Kimish Patel | f4a9216 | 2021-05-04 09:17:43 -0700 | [diff] [blame] | 591 | "torch/csrc/jit/serialization/source_range_serialization.cpp", |
| 592 | # Later we can split serialization and deserialization logic |
| 593 | # to have better separation within build and only build relevant parts. |
| 594 | "torch/csrc/jit/serialization/pickle.cpp", |
| 595 | "torch/csrc/jit/serialization/pickler.cpp", |
| 596 | "torch/csrc/jit/serialization/unpickler.cpp", |
| 597 | ] |
| 598 | |
Chen Lai | 14f7bf0 | 2021-02-21 01:41:55 -0800 | [diff] [blame] | 599 | # TODO: core_trainer_sources is not necessary for libtorch lite |
Pavithran Ramachandran | c2ceba8 | 2021-11-04 15:52:03 -0700 | [diff] [blame] | 600 | libtorch_lite_cmake_sources = sorted( |
| 601 | core_trainer_sources + |
| 602 | core_sources_common + |
| 603 | torch_unpickler_common + |
| 604 | torch_mobile_core, |
| 605 | ) |
Chen Lai | 14f7bf0 | 2021-02-21 01:41:55 -0800 | [diff] [blame] | 606 | |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 607 | libtorch_cmake_sources = libtorch_core_sources + libtorch_core_jit_sources |
| 608 | |
Nikita Shulga | d7fc05b | 2020-04-16 19:08:16 -0700 | [diff] [blame] | 609 | libtorch_extra_sources = libtorch_core_jit_sources + [ |
Jiakai Liu | 7f60642 | 2020-07-06 14:18:38 -0700 | [diff] [blame] | 610 | "torch/csrc/autograd/TraceTypeManual.cpp", |
Nikita Shulga | d7fc05b | 2020-04-16 19:08:16 -0700 | [diff] [blame] | 611 | "torch/csrc/autograd/VariableTypeManual.cpp", |
albanD | 73f009a | 2020-09-02 09:18:14 -0700 | [diff] [blame] | 612 | "torch/csrc/autograd/FunctionsManual.cpp", |
Nikita Shulga | d7fc05b | 2020-04-16 19:08:16 -0700 | [diff] [blame] | 613 | "torch/csrc/jit/api/module_save.cpp", |
| 614 | "torch/csrc/jit/codegen/fuser/cpu/fused_kernel.cpp", |
Jacob Szwejbka | e926360 | 2022-01-20 10:04:43 -0800 | [diff] [blame] | 615 | "torch/csrc/jit/mobile/compatibility/backport.cpp", |
| 616 | "torch/csrc/jit/mobile/compatibility/backport_manager.cpp", |
| 617 | "torch/csrc/jit/mobile/compatibility/model_compatibility.cpp", |
Kimish Patel | f4a9216 | 2021-05-04 09:17:43 -0700 | [diff] [blame] | 618 | # To be included for eager symbolication in lite interpreter |
| 619 | # when it is built in libtorch |
Tugsbayasgalan (Tugsuu) Manlaibaatar | e849c8b | 2022-01-27 18:27:28 -0800 | [diff] [blame] | 620 | "torch/csrc/jit/mobile/debug_info.cpp", |
| 621 | "torch/csrc/jit/mobile/function.cpp", |
Han Qi (qihqi) | 25eb7c3 | 2022-12-08 03:48:04 +0000 | [diff] [blame] | 622 | "torch/csrc/jit/mobile/flatbuffer_loader.cpp", |
Nikita Shulga | d7fc05b | 2020-04-16 19:08:16 -0700 | [diff] [blame] | 623 | "torch/csrc/jit/mobile/import.cpp", |
Ann Shan | dfe7d27 | 2020-07-23 14:23:30 -0700 | [diff] [blame] | 624 | "torch/csrc/jit/mobile/import_data.cpp", |
Tugsbayasgalan (Tugsuu) Manlaibaatar | e849c8b | 2022-01-27 18:27:28 -0800 | [diff] [blame] | 625 | "torch/csrc/jit/mobile/interpreter.cpp", |
| 626 | "torch/csrc/jit/mobile/module.cpp", |
| 627 | "torch/csrc/jit/mobile/observer.cpp", |
Martin Yuan | 30a7c76 | 2021-09-11 22:22:28 -0700 | [diff] [blame] | 628 | "torch/csrc/jit/mobile/parse_bytecode.cpp", |
Mengwei Liu | eaf85fa | 2021-09-17 12:57:48 -0700 | [diff] [blame] | 629 | "torch/csrc/jit/mobile/parse_operators.cpp", |
Kimish Patel | cfd18e1 | 2022-08-27 16:06:16 -0700 | [diff] [blame] | 630 | "torch/csrc/jit/mobile/quantization.cpp", |
Lunwen He | 73d5140 | 2021-05-14 12:53:18 -0700 | [diff] [blame] | 631 | "torch/csrc/jit/mobile/train/export_data.cpp", |
| 632 | "torch/csrc/jit/mobile/train/optim/sgd.cpp", |
| 633 | "torch/csrc/jit/mobile/train/random.cpp", |
| 634 | "torch/csrc/jit/mobile/train/sequential.cpp", |
Chen Lai | 9e4d60a | 2021-12-14 19:04:32 -0800 | [diff] [blame] | 635 | "torch/csrc/jit/mobile/upgrader_mobile.cpp", |
BowenBao | 08126c9 | 2020-08-30 18:33:47 -0700 | [diff] [blame] | 636 | "torch/csrc/jit/serialization/onnx.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 637 | "torch/csrc/jit/serialization/export.cpp", |
Tugsbayasgalan (Tugsuu) Manlaibaatar | e849c8b | 2022-01-27 18:27:28 -0800 | [diff] [blame] | 638 | "torch/csrc/jit/serialization/export_bytecode.cpp", |
| 639 | "torch/csrc/jit/serialization/export_module.cpp", |
Han Qi (qihqi) | 25eb7c3 | 2022-12-08 03:48:04 +0000 | [diff] [blame] | 640 | "torch/csrc/jit/serialization/flatbuffer_serializer.cpp", |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 641 | "torch/csrc/jit/serialization/import_legacy.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 642 | "torch/csrc/utils/byte_order.cpp", |
Peter Bell | 2639114 | 2021-01-05 17:15:37 -0800 | [diff] [blame] | 643 | "torch/csrc/utils/out_types.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 644 | ] |
| 645 | |
Nikita Shulga | 1571dd8 | 2020-06-30 13:42:01 -0700 | [diff] [blame] | 646 | def libtorch_sources(gencode_pattern = ":generate-code[{}]"): |
Han Qi (qihqi) | 25eb7c3 | 2022-12-08 03:48:04 +0000 | [diff] [blame] | 647 | return ( |
| 648 | libtorch_generated_sources(gencode_pattern) + libtorch_core_sources + libtorch_distributed_sources + libtorch_extra_sources |
| 649 | ) |
Nikita Shulga | 70d3616 | 2020-04-14 21:45:30 -0700 | [diff] [blame] | 650 | |
Nikita Shulga | 58551e5 | 2020-12-15 20:49:28 -0800 | [diff] [blame] | 651 | libtorch_cuda_core_sources = [ |
Will Constable | 4bbff92 | 2021-01-22 09:13:55 -0800 | [diff] [blame] | 652 | "torch/csrc/CudaIPCTypes.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 653 | "torch/csrc/cuda/comm.cpp", |
Zachary DeVito | 736adc0 | 2022-10-04 21:50:27 -0700 | [diff] [blame] | 654 | "torch/csrc/cuda/memory_snapshot.cpp", |
Mu-Chu Lee | eddce3c | 2023-11-14 14:08:25 -0800 | [diff] [blame] | 655 | "torch/csrc/inductor/aoti_model_container_runner_cuda.cpp", |
Bin Bao | 0f646b1 | 2023-09-16 16:46:26 +0000 | [diff] [blame] | 656 | "torch/csrc/inductor/aoti_torch/shim_cuda.cpp", |
Michael Suo | dbe850a | 2020-02-27 12:18:24 -0800 | [diff] [blame] | 657 | "torch/csrc/jit/codegen/fuser/cuda/fused_kernel.cpp", |
Taylor Robie | b8f14b7 | 2022-10-13 07:48:58 -0700 | [diff] [blame] | 658 | "torch/csrc/profiler/stubs/cuda.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 659 | "torch/csrc/autograd/functions/comm.cpp", |
David Berard | 8b1e496 | 2021-12-06 21:04:14 -0800 | [diff] [blame] | 660 | "torch/csrc/jit/passes/frozen_conv_add_relu_fusion_cuda.cpp", |
Mikhail Zolotukhin | 35e7efe | 2020-03-16 11:38:29 -0700 | [diff] [blame] | 661 | "torch/csrc/jit/tensorexpr/cuda_codegen.cpp", |
Nikitha Malgi | 12b73fd | 2020-12-29 20:22:19 -0800 | [diff] [blame] | 662 | "torch/csrc/jit/runtime/register_cuda_ops.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 663 | ] |
| 664 | |
Luca Wehrstedt | a178043 | 2021-06-15 02:00:08 -0700 | [diff] [blame] | 665 | # These files are the only ones that are supported on Windows. |
| 666 | libtorch_cuda_distributed_base_sources = [ |
Luca Wehrstedt | a016150 | 2021-06-24 12:37:29 -0700 | [diff] [blame] | 667 | "torch/csrc/distributed/c10d/reducer_cuda.cpp", |
Luca Wehrstedt | a178043 | 2021-06-15 02:00:08 -0700 | [diff] [blame] | 668 | ] |
| 669 | |
| 670 | # These files are only supported on Linux (and others) but not on Windows. |
| 671 | libtorch_cuda_distributed_extra_sources = [ |
Luca Wehrstedt | a016150 | 2021-06-24 12:37:29 -0700 | [diff] [blame] | 672 | "torch/csrc/distributed/c10d/NCCLUtils.cpp", |
| 673 | "torch/csrc/distributed/c10d/ProcessGroupNCCL.cpp", |
Terry Lam | 54bdaf7 | 2022-07-12 14:45:44 +0000 | [diff] [blame] | 674 | "torch/csrc/distributed/c10d/ProcessGroupUCC.cpp", |
| 675 | "torch/csrc/distributed/c10d/UCCTracing.cpp", |
| 676 | "torch/csrc/distributed/c10d/UCCUtils.cpp", |
Luca Wehrstedt | 08ce5ee | 2021-06-18 05:12:27 -0700 | [diff] [blame] | 677 | "torch/csrc/distributed/rpc/tensorpipe_cuda.cpp", |
Marjan Fariborz | 3b284ab | 2021-08-25 23:40:09 -0700 | [diff] [blame] | 678 | "torch/csrc/distributed/c10d/quantization/quantization_gpu.cu", |
Luca Wehrstedt | a178043 | 2021-06-15 02:00:08 -0700 | [diff] [blame] | 679 | ] |
| 680 | |
| 681 | libtorch_cuda_distributed_sources = libtorch_cuda_distributed_base_sources + libtorch_cuda_distributed_extra_sources |
| 682 | |
| 683 | libtorch_cuda_sources = libtorch_cuda_core_sources + libtorch_cuda_distributed_sources + [ |
Nikita Shulga | 58551e5 | 2020-12-15 20:49:28 -0800 | [diff] [blame] | 684 | "torch/csrc/cuda/nccl.cpp", |
| 685 | ] |
| 686 | |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 687 | torch_cpp_srcs = [ |
| 688 | "torch/csrc/api/src/cuda.cpp", # this just forwards stuff, no real CUDA |
| 689 | "torch/csrc/api/src/data/datasets/mnist.cpp", |
| 690 | "torch/csrc/api/src/data/samplers/distributed.cpp", |
| 691 | "torch/csrc/api/src/data/samplers/random.cpp", |
| 692 | "torch/csrc/api/src/data/samplers/sequential.cpp", |
| 693 | "torch/csrc/api/src/data/samplers/stream.cpp", |
| 694 | "torch/csrc/api/src/enum.cpp", |
Jiewen Tan | 641f6ef | 2021-07-19 23:15:51 -0700 | [diff] [blame] | 695 | "torch/csrc/api/src/imethod.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 696 | "torch/csrc/api/src/jit.cpp", |
Ramin Azarmehr | 234df29 | 2023-03-14 20:27:40 +0000 | [diff] [blame] | 697 | "torch/csrc/api/src/mps.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 698 | "torch/csrc/api/src/serialize.cpp", |
| 699 | "torch/csrc/api/src/nn/init.cpp", |
| 700 | "torch/csrc/api/src/nn/module.cpp", |
| 701 | "torch/csrc/api/src/nn/modules/_functions.cpp", |
| 702 | "torch/csrc/api/src/nn/modules/activation.cpp", |
Mansoor | e95657b | 2020-03-12 09:46:48 -0700 | [diff] [blame] | 703 | "torch/csrc/api/src/nn/modules/adaptive.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 704 | "torch/csrc/api/src/nn/modules/batchnorm.cpp", |
| 705 | "torch/csrc/api/src/nn/modules/normalization.cpp", |
| 706 | "torch/csrc/api/src/nn/modules/instancenorm.cpp", |
| 707 | "torch/csrc/api/src/nn/modules/conv.cpp", |
| 708 | "torch/csrc/api/src/nn/modules/dropout.cpp", |
| 709 | "torch/csrc/api/src/nn/modules/distance.cpp", |
| 710 | "torch/csrc/api/src/nn/modules/embedding.cpp", |
| 711 | "torch/csrc/api/src/nn/modules/fold.cpp", |
| 712 | "torch/csrc/api/src/nn/modules/linear.cpp", |
| 713 | "torch/csrc/api/src/nn/modules/loss.cpp", |
| 714 | "torch/csrc/api/src/nn/modules/padding.cpp", |
| 715 | "torch/csrc/api/src/nn/modules/pixelshuffle.cpp", |
| 716 | "torch/csrc/api/src/nn/modules/pooling.cpp", |
| 717 | "torch/csrc/api/src/nn/modules/rnn.cpp", |
| 718 | "torch/csrc/api/src/nn/modules/upsampling.cpp", |
lixinyu | 98de150 | 2020-08-07 11:48:15 -0700 | [diff] [blame] | 719 | "torch/csrc/api/src/nn/modules/transformer.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 720 | "torch/csrc/api/src/nn/modules/container/functional.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 721 | "torch/csrc/api/src/nn/options/activation.cpp", |
Mansoor | e95657b | 2020-03-12 09:46:48 -0700 | [diff] [blame] | 722 | "torch/csrc/api/src/nn/options/adaptive.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 723 | "torch/csrc/api/src/nn/options/batchnorm.cpp", |
| 724 | "torch/csrc/api/src/nn/options/conv.cpp", |
| 725 | "torch/csrc/api/src/nn/options/dropout.cpp", |
| 726 | "torch/csrc/api/src/nn/options/instancenorm.cpp", |
| 727 | "torch/csrc/api/src/nn/options/linear.cpp", |
| 728 | "torch/csrc/api/src/nn/options/normalization.cpp", |
| 729 | "torch/csrc/api/src/nn/options/embedding.cpp", |
| 730 | "torch/csrc/api/src/nn/options/padding.cpp", |
| 731 | "torch/csrc/api/src/nn/options/pooling.cpp", |
| 732 | "torch/csrc/api/src/nn/options/rnn.cpp", |
| 733 | "torch/csrc/api/src/nn/options/vision.cpp", |
lixinyu | 98de150 | 2020-08-07 11:48:15 -0700 | [diff] [blame] | 734 | "torch/csrc/api/src/nn/options/transformer.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 735 | "torch/csrc/api/src/optim/adagrad.cpp", |
| 736 | "torch/csrc/api/src/optim/adam.cpp", |
Sotiris Lamprinidis | 41f2dbd | 2020-06-18 15:26:21 -0700 | [diff] [blame] | 737 | "torch/csrc/api/src/optim/adamw.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 738 | "torch/csrc/api/src/optim/lbfgs.cpp", |
| 739 | "torch/csrc/api/src/optim/optimizer.cpp", |
| 740 | "torch/csrc/api/src/optim/rmsprop.cpp", |
| 741 | "torch/csrc/api/src/optim/serialize.cpp", |
| 742 | "torch/csrc/api/src/optim/sgd.cpp", |
James Butterworth | 37ab711 | 2021-03-10 23:07:43 -0800 | [diff] [blame] | 743 | "torch/csrc/api/src/optim/schedulers/lr_scheduler.cpp", |
| 744 | "torch/csrc/api/src/optim/schedulers/step_lr.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 745 | "torch/csrc/api/src/serialize/input-archive.cpp", |
| 746 | "torch/csrc/api/src/serialize/output-archive.cpp", |
| 747 | ] |
| 748 | |
Yujun Zhao | 461014d | 2020-06-25 10:52:18 -0700 | [diff] [blame] | 749 | libtorch_python_cuda_core_sources = [ |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 750 | "torch/csrc/cuda/Event.cpp", |
| 751 | "torch/csrc/cuda/Module.cpp", |
Yujun Zhao | 461014d | 2020-06-25 10:52:18 -0700 | [diff] [blame] | 752 | "torch/csrc/cuda/python_comm.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 753 | "torch/csrc/cuda/Stream.cpp", |
Michael Carilli | c068180 | 2020-12-14 10:49:57 -0800 | [diff] [blame] | 754 | "torch/csrc/cuda/Graph.cpp", |
Peter Bell | 5fc5cf6 | 2020-03-11 07:15:03 -0700 | [diff] [blame] | 755 | "torch/csrc/cuda/shared/cudart.cpp", |
Peter Bell | 5fc5cf6 | 2020-03-11 07:15:03 -0700 | [diff] [blame] | 756 | "torch/csrc/cuda/shared/nvtx.cpp", |
Yujun Zhao | 461014d | 2020-06-25 10:52:18 -0700 | [diff] [blame] | 757 | "torch/csrc/cuda/utils.cpp", |
Emilio Castillo | c9d4390 | 2022-11-23 17:54:33 +0000 | [diff] [blame] | 758 | "torch/csrc/cuda/CUDAPluggableAllocator.cpp", |
Yujun Zhao | 461014d | 2020-06-25 10:52:18 -0700 | [diff] [blame] | 759 | ] |
| 760 | |
| 761 | libtorch_python_cuda_sources = libtorch_python_cuda_core_sources + [ |
| 762 | "torch/csrc/cuda/python_nccl.cpp", |
| 763 | "torch/csrc/cuda/shared/cudnn.cpp", |
| 764 | "torch/csrc/cuda/Tensor.cpp", |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 765 | ] |
| 766 | |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 767 | libtorch_python_core_sources = [ |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 768 | "torch/csrc/DataLoader.cpp", |
| 769 | "torch/csrc/Device.cpp", |
| 770 | "torch/csrc/Dtype.cpp", |
| 771 | "torch/csrc/DynamicTypes.cpp", |
| 772 | "torch/csrc/Exceptions.cpp", |
| 773 | "torch/csrc/Generator.cpp", |
| 774 | "torch/csrc/Layout.cpp", |
| 775 | "torch/csrc/MemoryFormat.cpp", |
| 776 | "torch/csrc/QScheme.cpp", |
| 777 | "torch/csrc/Module.cpp", |
Kurt Mohler | 4d9920f | 2023-01-24 13:20:28 -0600 | [diff] [blame] | 778 | "torch/csrc/PyInterpreter.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 779 | "torch/csrc/python_dimname.cpp", |
| 780 | "torch/csrc/Size.cpp", |
| 781 | "torch/csrc/Storage.cpp", |
Kurt Mohler | 272193d | 2022-06-01 19:00:58 +0000 | [diff] [blame] | 782 | "torch/csrc/StorageMethods.cpp", |
| 783 | "torch/csrc/StorageSharing.cpp", |
chengjun | 5741de8 | 2020-10-13 09:13:00 -0700 | [diff] [blame] | 784 | "torch/csrc/Stream.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 785 | "torch/csrc/TypeInfo.cpp", |
| 786 | "torch/csrc/api/src/python/init.cpp", |
| 787 | "torch/csrc/autograd/functions/init.cpp", |
| 788 | "torch/csrc/autograd/init.cpp", |
Taylor Robie | 33e9a0b | 2021-11-15 23:30:05 -0800 | [diff] [blame] | 789 | "torch/csrc/autograd/profiler_python.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 790 | "torch/csrc/autograd/python_anomaly_mode.cpp", |
Victor Quach | ee5a97d | 2021-07-15 08:07:56 -0700 | [diff] [blame] | 791 | "torch/csrc/autograd/python_saved_variable_hooks.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 792 | "torch/csrc/autograd/python_cpp_function.cpp", |
| 793 | "torch/csrc/autograd/python_engine.cpp", |
| 794 | "torch/csrc/autograd/python_function.cpp", |
| 795 | "torch/csrc/autograd/python_hook.cpp", |
| 796 | "torch/csrc/autograd/python_legacy_variable.cpp", |
Antoni Viros i Martin | c77368d | 2022-11-08 00:03:14 +0000 | [diff] [blame] | 797 | "torch/csrc/autograd/python_nested_functions_manual.cpp", |
Peter Bell | 44ede71 | 2021-08-25 15:05:14 -0700 | [diff] [blame] | 798 | "torch/csrc/autograd/python_torch_functions_manual.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 799 | "torch/csrc/autograd/python_variable.cpp", |
| 800 | "torch/csrc/autograd/python_variable_indexing.cpp", |
Jason Ansel | c902b84 | 2023-07-23 17:17:29 -0700 | [diff] [blame] | 801 | "torch/csrc/dynamo/python_compiled_autograd.cpp", |
David Berard | 06b1737 | 2023-09-01 17:32:57 -0700 | [diff] [blame] | 802 | "torch/csrc/dynamo/cpp_shim.cpp", |
William Wen | 785676c | 2023-04-25 22:03:28 +0000 | [diff] [blame] | 803 | "torch/csrc/dynamo/cpython_defs.c", |
Jason Ansel | f1fdb6e | 2022-10-11 23:01:21 +0000 | [diff] [blame] | 804 | "torch/csrc/dynamo/eval_frame.c", |
| 805 | "torch/csrc/dynamo/guards.cpp", |
| 806 | "torch/csrc/dynamo/init.cpp", |
Richard Zou | 5e5c319 | 2022-09-22 06:56:40 -0700 | [diff] [blame] | 807 | "torch/csrc/functorch/init.cpp", |
Ramin Azarmehr | bdd8f51 | 2023-02-12 21:22:28 +0000 | [diff] [blame] | 808 | "torch/csrc/mps/Module.cpp", |
eellison | d5df055 | 2020-05-07 18:13:35 -0700 | [diff] [blame] | 809 | "torch/csrc/jit/backends/backend_init.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 810 | "torch/csrc/jit/python/init.cpp", |
| 811 | "torch/csrc/jit/passes/onnx.cpp", |
| 812 | "torch/csrc/jit/passes/onnx/cast_all_constant_to_floating.cpp", |
BowenBao | cc79274 | 2022-02-11 13:58:46 -0800 | [diff] [blame] | 813 | "torch/csrc/jit/passes/onnx/deduplicate_initializers.cpp", |
Ksenija Stanojevic | af5d0bf | 2020-07-22 14:57:23 -0700 | [diff] [blame] | 814 | "torch/csrc/jit/passes/onnx/eval_peephole.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 815 | "torch/csrc/jit/passes/onnx/constant_fold.cpp", |
BowenBao | 7f17058 | 2021-03-12 02:42:06 -0800 | [diff] [blame] | 816 | "torch/csrc/jit/passes/onnx/constant_map.cpp", |
Ksenija Stanojevic | e845b0a | 2020-08-11 20:29:12 -0700 | [diff] [blame] | 817 | "torch/csrc/jit/passes/onnx/eliminate_unused_items.cpp", |
BowenBao | 8427595 | 2020-08-03 22:31:54 -0700 | [diff] [blame] | 818 | "torch/csrc/jit/passes/onnx/fixup_onnx_controlflow.cpp", |
neginraoof | 15bc21c | 2020-12-03 23:05:43 -0800 | [diff] [blame] | 819 | "torch/csrc/jit/passes/onnx/list_model_parameters.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 820 | "torch/csrc/jit/passes/onnx/function_substitution.cpp", |
| 821 | "torch/csrc/jit/passes/onnx/helper.cpp", |
| 822 | "torch/csrc/jit/passes/onnx/peephole.cpp", |
BowenBao | a6c8730 | 2020-08-06 20:30:41 -0700 | [diff] [blame] | 823 | "torch/csrc/jit/passes/onnx/preprocess_for_onnx.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 824 | "torch/csrc/jit/passes/onnx/prepare_division_for_onnx.cpp", |
| 825 | "torch/csrc/jit/passes/onnx/scalar_type_analysis.cpp", |
| 826 | "torch/csrc/jit/passes/onnx/unpack_quantized_weights.cpp", |
neginraoof | 3d7c22a | 2020-09-04 15:19:37 -0700 | [diff] [blame] | 827 | "torch/csrc/jit/passes/onnx/remove_inplace_ops_for_onnx.cpp", |
BowenBao | 08126c9 | 2020-08-30 18:33:47 -0700 | [diff] [blame] | 828 | "torch/csrc/jit/passes/onnx/shape_type_inference.cpp", |
Nikita Shulga | 53a163a | 2021-10-22 13:36:41 -0700 | [diff] [blame] | 829 | "torch/csrc/jit/passes/onnx/function_extraction.cpp", |
BowenBao | 54a6942 | 2022-03-17 12:32:51 -0700 | [diff] [blame] | 830 | "torch/csrc/jit/passes/onnx/onnx_log.cpp", |
BowenBao | daca0ee | 2022-08-22 10:14:36 -0700 | [diff] [blame] | 831 | "torch/csrc/jit/passes/onnx/naming.cpp", |
Luca Wehrstedt | 1ac05cf | 2020-12-19 11:01:55 -0800 | [diff] [blame] | 832 | "torch/csrc/jit/python/pybind_utils.cpp", |
shubhambhokare1 | 95d8738 | 2022-08-03 23:30:19 +0000 | [diff] [blame] | 833 | "torch/csrc/jit/passes/onnx/pattern_conversion/autograd_function_process.cpp", |
BowenBao | 3f9c803 | 2021-03-11 10:25:12 -0800 | [diff] [blame] | 834 | "torch/csrc/jit/passes/onnx/pattern_conversion/common.cpp", |
| 835 | "torch/csrc/jit/passes/onnx/pattern_conversion/pattern_encapsulation.cpp", |
| 836 | "torch/csrc/jit/passes/onnx/pattern_conversion/pattern_conversion.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 837 | "torch/csrc/jit/python/python_arg_flatten.cpp", |
| 838 | "torch/csrc/jit/python/python_custom_class.cpp", |
Meghan Lele | b14c320 | 2021-05-27 10:24:11 -0700 | [diff] [blame] | 839 | "torch/csrc/jit/python/python_dict.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 840 | "torch/csrc/jit/python/python_interpreter.cpp", |
| 841 | "torch/csrc/jit/python/python_ir.cpp", |
Meghan Lele | 4a2e8b5 | 2021-07-01 20:27:02 -0700 | [diff] [blame] | 842 | "torch/csrc/jit/python/python_list.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 843 | "torch/csrc/jit/python/python_tracer.cpp", |
| 844 | "torch/csrc/jit/python/script_init.cpp", |
| 845 | "torch/csrc/jit/frontend/concrete_module_type.cpp", |
Zhengxu Chen | 8176ab6 | 2021-04-16 15:45:49 -0700 | [diff] [blame] | 846 | "torch/csrc/jit/frontend/tree_views.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 847 | "torch/csrc/jit/python/python_sugared_value.cpp", |
| 848 | "torch/csrc/jit/python/python_tree_views.cpp", |
Bram Wasti | ada8404 | 2020-08-12 13:02:29 -0700 | [diff] [blame] | 849 | "torch/csrc/jit/runtime/static/init.cpp", |
Mikhail Zolotukhin | e9dc8fc | 2021-01-14 21:00:53 -0800 | [diff] [blame] | 850 | "torch/csrc/jit/tensorexpr/tensorexpr_init.cpp", |
Tristan Rice | bfe1abd | 2022-01-12 13:33:49 -0800 | [diff] [blame] | 851 | "torch/csrc/monitor/python_init.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 852 | "torch/csrc/multiprocessing/init.cpp", |
| 853 | "torch/csrc/onnx/init.cpp", |
Taylor Robie | 1fa9a37 | 2022-08-18 17:51:32 -0700 | [diff] [blame] | 854 | "torch/csrc/profiler/python/init.cpp", |
Zachary DeVito | e74f70d | 2023-03-15 15:14:10 -0700 | [diff] [blame] | 855 | "torch/csrc/profiler/python/combined_traceback.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 856 | "torch/csrc/serialization.cpp", |
| 857 | "torch/csrc/tensor/python_tensor.cpp", |
| 858 | "torch/csrc/utils/init.cpp", |
| 859 | "torch/csrc/utils/throughput_benchmark.cpp", |
| 860 | "torch/csrc/utils.cpp", |
| 861 | "torch/csrc/utils/cuda_lazy_init.cpp", |
| 862 | "torch/csrc/utils/invalid_arguments.cpp", |
Antoni Viros i Martin | c77368d | 2022-11-08 00:03:14 +0000 | [diff] [blame] | 863 | "torch/csrc/utils/nested.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 864 | "torch/csrc/utils/object_ptr.cpp", |
| 865 | "torch/csrc/utils/python_arg_parser.cpp", |
| 866 | "torch/csrc/utils/python_dispatch.cpp", |
Edward Z. Yang | 1ff5222 | 2022-10-27 13:49:11 -0700 | [diff] [blame] | 867 | "torch/csrc/utils/python_symnode.cpp", |
Edward Z. Yang | d3c01c7 | 2022-10-28 17:20:10 -0400 | [diff] [blame] | 868 | "torch/csrc/utils/pybind.cpp", |
Kurt Mohler | 4c5e435 | 2023-09-12 22:26:05 +0000 | [diff] [blame] | 869 | "torch/csrc/utils/pyobject_preservation.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 870 | "torch/csrc/utils/structseq.cpp", |
| 871 | "torch/csrc/utils/tensor_apply.cpp", |
| 872 | "torch/csrc/utils/tensor_dtypes.cpp", |
| 873 | "torch/csrc/utils/tensor_layouts.cpp", |
| 874 | "torch/csrc/utils/tensor_memoryformats.cpp", |
| 875 | "torch/csrc/utils/tensor_qschemes.cpp", |
| 876 | "torch/csrc/utils/tensor_list.cpp", |
| 877 | "torch/csrc/utils/tensor_new.cpp", |
| 878 | "torch/csrc/utils/tensor_numpy.cpp", |
| 879 | "torch/csrc/utils/tensor_types.cpp", |
Hameer Abbasi | 3d46e02 | 2020-08-05 20:39:27 -0700 | [diff] [blame] | 880 | "torch/csrc/utils/disable_torch_function.cpp", |
Jing Xu | 0e95746 | 2022-07-27 23:29:35 +0000 | [diff] [blame] | 881 | "torch/csrc/utils/verbose.cpp", |
leslie-fang-intel | 9832cfb | 2023-06-19 14:52:39 +0800 | [diff] [blame] | 882 | "torch/csrc/cpu/Module.cpp", |
Will Constable | 328cfd5 | 2022-02-11 09:54:34 -0800 | [diff] [blame] | 883 | ] + lazy_tensor_core_python_sources |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 884 | |
gunandrose4u | f07ac6a | 2020-09-25 12:35:42 -0700 | [diff] [blame] | 885 | libtorch_python_distributed_core_sources = [ |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 886 | "torch/csrc/distributed/c10d/init.cpp", |
Luca Wehrstedt | a016150 | 2021-06-24 12:37:29 -0700 | [diff] [blame] | 887 | "torch/csrc/distributed/c10d/python_comm_hook.cpp", |
gunandrose4u | f07ac6a | 2020-09-25 12:35:42 -0700 | [diff] [blame] | 888 | ] |
| 889 | |
| 890 | libtorch_python_distributed_sources = libtorch_python_distributed_core_sources + [ |
| 891 | "torch/csrc/distributed/autograd/init.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 892 | "torch/csrc/distributed/rpc/init.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 893 | "torch/csrc/distributed/rpc/py_rref.cpp", |
| 894 | "torch/csrc/distributed/rpc/python_functions.cpp", |
| 895 | "torch/csrc/distributed/rpc/python_rpc_handler.cpp", |
| 896 | "torch/csrc/distributed/rpc/request_callback_impl.cpp", |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 897 | "torch/csrc/distributed/rpc/testing/init.cpp", |
| 898 | "torch/csrc/distributed/rpc/unpickled_python_call.cpp", |
| 899 | "torch/csrc/distributed/rpc/unpickled_python_remote_call.cpp", |
| 900 | "torch/csrc/jit/runtime/register_distributed_ops.cpp", |
| 901 | ] |
| 902 | |
Nikita Shulga | 1571dd8 | 2020-06-30 13:42:01 -0700 | [diff] [blame] | 903 | def glob_libtorch_python_sources(gencode_pattern = ":generate-code[{}]"): |
| 904 | _libtorch_python_sources = [gencode_pattern.format(name) for name in [ |
mikey dagitses | 096ff0e | 2022-05-04 06:48:20 -0700 | [diff] [blame] | 905 | "torch/csrc/autograd/generated/python_functions_0.cpp", |
| 906 | "torch/csrc/autograd/generated/python_functions_1.cpp", |
| 907 | "torch/csrc/autograd/generated/python_functions_2.cpp", |
| 908 | "torch/csrc/autograd/generated/python_functions_3.cpp", |
| 909 | "torch/csrc/autograd/generated/python_functions_4.cpp", |
Mikayla Gawarecki | e217b30 | 2022-09-12 04:03:49 +0000 | [diff] [blame] | 910 | "torch/csrc/autograd/generated/python_nested_functions.cpp", |
mikey dagitses | 096ff0e | 2022-05-04 06:48:20 -0700 | [diff] [blame] | 911 | "torch/csrc/autograd/generated/python_nn_functions.cpp", |
| 912 | "torch/csrc/autograd/generated/python_fft_functions.cpp", |
| 913 | "torch/csrc/autograd/generated/python_linalg_functions.cpp", |
anjali411 | 38350ac | 2022-06-10 21:48:56 +0000 | [diff] [blame] | 914 | "torch/csrc/autograd/generated/python_enum_tag.cpp", |
mikey dagitses | 096ff0e | 2022-05-04 06:48:20 -0700 | [diff] [blame] | 915 | "torch/csrc/autograd/generated/python_return_types.cpp", |
| 916 | "torch/csrc/autograd/generated/python_sparse_functions.cpp", |
| 917 | "torch/csrc/autograd/generated/python_special_functions.cpp", |
| 918 | "torch/csrc/autograd/generated/python_torch_functions_0.cpp", |
| 919 | "torch/csrc/autograd/generated/python_torch_functions_1.cpp", |
| 920 | "torch/csrc/autograd/generated/python_torch_functions_2.cpp", |
| 921 | "torch/csrc/autograd/generated/python_variable_methods.cpp", |
Nikita Shulga | 1571dd8 | 2020-06-30 13:42:01 -0700 | [diff] [blame] | 922 | ]] |
Michael Suo | 0bde610 | 2020-02-20 16:46:16 -0800 | [diff] [blame] | 923 | |
Nikita Shulga | d294c06 | 2020-04-27 16:07:01 -0700 | [diff] [blame] | 924 | _libtorch_python_sources.extend(libtorch_python_core_sources) |
| 925 | _libtorch_python_sources.extend(libtorch_python_distributed_sources) |
| 926 | |
Alexander Fix | 3da67ce | 2020-04-06 18:22:32 -0700 | [diff] [blame] | 927 | return _libtorch_python_sources |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 928 | |
Nikita Shulga | 6cef594 | 2022-12-31 00:02:43 +0000 | [diff] [blame] | 929 | # List of non-globed source used to build ATen core internally |
| 930 | aten_cpu_non_globed_sources = [ |
| 931 | "aten/src/ATen/detail/CUDAHooksInterface.cpp", |
| 932 | "aten/src/ATen/detail/HIPHooksInterface.cpp", |
| 933 | "aten/src/ATen/detail/MPSHooksInterface.cpp", |
| 934 | "aten/src/ATen/detail/ORTHooksInterface.cpp", |
shibo19 | 7047d13 | 2023-07-29 12:56:03 +0000 | [diff] [blame] | 935 | "aten/src/ATen/detail/PrivateUse1HooksInterface.cpp", |
Xunsong, Huang | b053a0f | 2023-03-10 12:17:14 +0000 | [diff] [blame] | 936 | "aten/src/ATen/detail/XPUHooksInterface.cpp", |
Richard Li | c523d7d | 2023-04-26 23:00:34 +0000 | [diff] [blame] | 937 | "aten/src/ATen/detail/MTIAHooksInterface.cpp", |
George White | f4796df | 2023-10-06 21:36:10 +0000 | [diff] [blame] | 938 | "aten/src/ATen/detail/IPUHooksInterface.cpp", |
Nikita Shulga | 6cef594 | 2022-12-31 00:02:43 +0000 | [diff] [blame] | 939 | "aten/src/ATen/record_function.cpp", |
| 940 | "aten/src/ATen/Dispatch.cpp", |
| 941 | "aten/src/ATen/SequenceNumber.cpp", |
| 942 | ] |
| 943 | |
| 944 | aten_cpu_non_globed_headers = [ |
| 945 | "aten/src/ATen/CPUGeneratorImpl.h", |
| 946 | "aten/src/ATen/NumericUtils.h", |
| 947 | "aten/src/ATen/detail/CUDAHooksInterface.h", |
| 948 | "aten/src/ATen/detail/MPSHooksInterface.h", |
| 949 | "aten/src/ATen/detail/HIPHooksInterface.h", |
| 950 | "aten/src/ATen/detail/ORTHooksInterface.h", |
shibo19 | 7047d13 | 2023-07-29 12:56:03 +0000 | [diff] [blame] | 951 | "aten/src/ATen/detail/PrivateUse1HooksInterface.h", |
Xunsong, Huang | b053a0f | 2023-03-10 12:17:14 +0000 | [diff] [blame] | 952 | "aten/src/ATen/detail/XPUHooksInterface.h", |
Richard Li | c523d7d | 2023-04-26 23:00:34 +0000 | [diff] [blame] | 953 | "aten/src/ATen/detail/MTIAHooksInterface.h", |
George White | f4796df | 2023-10-06 21:36:10 +0000 | [diff] [blame] | 954 | "aten/src/ATen/detail/IPUHooksInterface.h", |
Nikita Shulga | 6cef594 | 2022-12-31 00:02:43 +0000 | [diff] [blame] | 955 | ] |
| 956 | |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 957 | aten_cpu_source_non_codegen_list = [ |
Peter Bell | 6408cbd | 2021-06-03 11:42:14 -0700 | [diff] [blame] | 958 | "aten/src/ATen/AccumulateType.cpp", |
Richard Zou | 4b10534 | 2022-12-07 07:48:41 -0800 | [diff] [blame] | 959 | "aten/src/ATen/LegacyBatchedTensorImpl.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 960 | "aten/src/ATen/CPUGeneratorImpl.cpp", |
| 961 | "aten/src/ATen/Context.cpp", |
| 962 | "aten/src/ATen/DLConvertor.cpp", |
Peter Bell | 6f4c491 | 2022-01-18 15:57:30 -0800 | [diff] [blame] | 963 | "aten/src/ATen/EmptyTensor.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 964 | "aten/src/ATen/ExpandUtils.cpp", |
Elias Ellison | d881b29 | 2023-04-24 17:08:40 +0000 | [diff] [blame] | 965 | "aten/src/ATen/CachedTensorUtils.cpp", |
Brian Hirsh | 0032fa7 | 2021-10-28 10:43:11 -0700 | [diff] [blame] | 966 | "aten/src/ATen/FunctionalInverses.cpp", |
| 967 | "aten/src/ATen/FunctionalStorageImpl.cpp", |
| 968 | "aten/src/ATen/FunctionalTensorWrapper.cpp", |
Brian Hirsh | 4a2bbc6 | 2021-10-29 21:39:29 -0700 | [diff] [blame] | 969 | "aten/src/ATen/FunctionalizeFallbackKernel.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 970 | "aten/src/ATen/MemoryOverlap.cpp", |
PyTorch MergeBot | f68d6e8 | 2023-10-13 17:57:53 +0000 | [diff] [blame] | 971 | "aten/src/ATen/MapAllocator.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 972 | "aten/src/ATen/NamedTensorUtils.cpp", |
Christian Puhrsch | 484c0de | 2022-03-02 07:29:19 -0800 | [diff] [blame] | 973 | "aten/src/ATen/NestedTensorImpl.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 974 | "aten/src/ATen/ParallelCommon.cpp", |
| 975 | "aten/src/ATen/ParallelNative.cpp", |
| 976 | "aten/src/ATen/ParallelNativeTBB.cpp", |
| 977 | "aten/src/ATen/ParallelOpenMP.cpp", |
| 978 | "aten/src/ATen/ParallelThreadPoolNative.cpp", |
Edward Z. Yang | 3f108a5 | 2022-04-01 13:15:24 -0700 | [diff] [blame] | 979 | "aten/src/ATen/PythonTorchFunctionTLS.cpp", |
Elias Ellison | 70f4b35 | 2023-01-23 21:03:42 +0000 | [diff] [blame] | 980 | "aten/src/ATen/ThreadLocalPythonObjects.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 981 | "aten/src/ATen/ScalarOps.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 982 | "aten/src/ATen/SparseTensorImpl.cpp", |
Sameer Deshmukh | 5fb1142 | 2021-04-12 10:07:56 -0700 | [diff] [blame] | 983 | "aten/src/ATen/SparseCsrTensorImpl.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 984 | "aten/src/ATen/TensorGeometry.cpp", |
| 985 | "aten/src/ATen/TensorIndexing.cpp", |
| 986 | "aten/src/ATen/TensorMeta.cpp", |
| 987 | "aten/src/ATen/TensorNames.cpp", |
| 988 | "aten/src/ATen/TensorUtils.cpp", |
| 989 | "aten/src/ATen/ThreadLocalState.cpp", |
Richard Zou | 698c35e | 2021-12-02 09:24:19 -0800 | [diff] [blame] | 990 | "aten/src/ATen/FuncTorchTLS.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 991 | "aten/src/ATen/Utils.cpp", |
| 992 | "aten/src/ATen/Version.cpp", |
Richard Zou | 4b10534 | 2022-12-07 07:48:41 -0800 | [diff] [blame] | 993 | "aten/src/ATen/LegacyVmapMode.cpp", |
| 994 | "aten/src/ATen/LegacyVmapTransforms.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 995 | "aten/src/ATen/core/BackendSelectFallbackKernel.cpp", |
| 996 | "aten/src/ATen/core/DeprecatedTypeProperties.cpp", |
| 997 | "aten/src/ATen/core/DeprecatedTypePropertiesRegistry.cpp", |
| 998 | "aten/src/ATen/core/Dict.cpp", |
| 999 | "aten/src/ATen/core/Dimname.cpp", |
| 1000 | "aten/src/ATen/core/Formatting.cpp", |
goldenxuett | e3a8709 | 2022-07-11 15:01:50 -0700 | [diff] [blame] | 1001 | "aten/src/ATen/core/function_schema.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1002 | "aten/src/ATen/core/Generator.cpp", |
rzou | 8124a6c | 2023-09-21 14:04:16 -0700 | [diff] [blame] | 1003 | "aten/src/ATen/core/PythonOpRegistrationTrampoline.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1004 | "aten/src/ATen/core/List.cpp", |
| 1005 | "aten/src/ATen/core/NamedTensor.cpp", |
| 1006 | "aten/src/ATen/core/Tensor.cpp", |
| 1007 | "aten/src/ATen/core/VariableFallbackKernel.cpp", |
| 1008 | "aten/src/ATen/core/VariableHooksInterface.cpp", |
Victor Bittorf | 52f1a07 | 2021-04-15 15:56:00 -0700 | [diff] [blame] | 1009 | "aten/src/ATen/core/Vitals.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1010 | "aten/src/ATen/core/boxing/KernelFunction.cpp", |
| 1011 | "aten/src/ATen/core/custom_class.cpp", |
| 1012 | "aten/src/ATen/core/dispatch/DispatchKeyExtractor.cpp", |
| 1013 | "aten/src/ATen/core/dispatch/Dispatcher.cpp", |
| 1014 | "aten/src/ATen/core/dispatch/ObservedOperators.cpp", |
| 1015 | "aten/src/ATen/core/dispatch/OperatorEntry.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1016 | "aten/src/ATen/core/interned_strings.cpp", |
| 1017 | "aten/src/ATen/core/ivalue.cpp", |
| 1018 | "aten/src/ATen/core/library.cpp", |
| 1019 | "aten/src/ATen/core/op_registration/infer_schema.cpp", |
| 1020 | "aten/src/ATen/core/op_registration/op_registration.cpp", |
| 1021 | "aten/src/ATen/core/operator_name.cpp", |
Edward Z. Yang | 2d8f091 | 2022-08-15 06:56:28 -0700 | [diff] [blame] | 1022 | "aten/src/ATen/core/TorchDispatchUtils.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1023 | "aten/src/ATen/core/register_symbols.cpp", |
Zhengxu Chen | 39f65fe | 2021-12-16 13:06:08 -0800 | [diff] [blame] | 1024 | "aten/src/ATen/core/class_type.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1025 | "aten/src/ATen/core/type.cpp", |
Zhengxu Chen | e7634f8 | 2022-01-11 13:47:58 -0800 | [diff] [blame] | 1026 | "aten/src/ATen/core/type_factory.cpp", |
Zhengxu Chen | 649dda9 | 2022-01-07 11:19:15 -0800 | [diff] [blame] | 1027 | "aten/src/ATen/core/dynamic_type.cpp", |
Zhengxu Chen | 3f3eae6 | 2022-01-07 18:32:35 -0800 | [diff] [blame] | 1028 | "aten/src/ATen/core/tensor_type.cpp", |
Zhengxu Chen | bc026c0 | 2022-01-07 11:19:15 -0800 | [diff] [blame] | 1029 | "aten/src/ATen/core/union_type.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1030 | "aten/src/ATen/cpu/FlushDenormal.cpp", |
| 1031 | "aten/src/ATen/detail/CPUGuardImpl.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1032 | "aten/src/ATen/metal/Context.cpp", |
| 1033 | "aten/src/ATen/native/AutogradComposite.cpp", |
Michael Voznesensky | cd7408e | 2022-09-19 20:48:09 +0000 | [diff] [blame] | 1034 | "aten/src/ATen/native/ComparisonUtils.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1035 | "aten/src/ATen/native/DispatchStub.cpp", |
| 1036 | "aten/src/ATen/native/UpSample.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1037 | "aten/src/ATen/native/mkldnn/BinaryOps.cpp", |
| 1038 | "aten/src/ATen/native/mkldnn/Conv.cpp", |
chunyuan-w | 693a8dd | 2022-08-10 21:46:51 +0000 | [diff] [blame] | 1039 | "aten/src/ATen/native/mkldnn/ConvPrepack.cpp", |
Masaya, Kato | 473d193 | 2021-04-08 06:34:10 -0700 | [diff] [blame] | 1040 | "aten/src/ATen/native/mkldnn/Copy.cpp", |
yanbing-j | c7a7c2b | 2021-08-03 06:50:34 -0700 | [diff] [blame] | 1041 | "aten/src/ATen/native/mkldnn/Gelu.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1042 | "aten/src/ATen/native/mkldnn/IDeepRegistration.cpp", |
| 1043 | "aten/src/ATen/native/mkldnn/Linear.cpp", |
| 1044 | "aten/src/ATen/native/mkldnn/MKLDNNCommon.cpp", |
| 1045 | "aten/src/ATen/native/mkldnn/MKLDNNConversions.cpp", |
| 1046 | "aten/src/ATen/native/mkldnn/MkldnnTensorMath.cpp", |
| 1047 | "aten/src/ATen/native/mkldnn/Normalization.cpp", |
chunyuan-w | 693a8dd | 2022-08-10 21:46:51 +0000 | [diff] [blame] | 1048 | "aten/src/ATen/native/mkldnn/OpContext.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1049 | "aten/src/ATen/native/mkldnn/Pooling.cpp", |
yanbing-j | cd33e41 | 2022-05-10 17:29:11 +0000 | [diff] [blame] | 1050 | "aten/src/ATen/native/mkldnn/Prelu.cpp", |
chunyuan-w | 693a8dd | 2022-08-10 21:46:51 +0000 | [diff] [blame] | 1051 | "aten/src/ATen/native/mkldnn/RegisterMkldnnOpContextClass.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1052 | "aten/src/ATen/native/mkldnn/Relu.cpp", |
yanbing-j | 94a7c01 | 2023-01-18 04:41:18 +0000 | [diff] [blame] | 1053 | "aten/src/ATen/native/mkldnn/RNN.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1054 | "aten/src/ATen/native/mkldnn/SoftMax.cpp", |
| 1055 | "aten/src/ATen/native/mkldnn/TensorFactories.cpp", |
| 1056 | "aten/src/ATen/native/mkldnn/TensorShape.cpp", |
| 1057 | "aten/src/ATen/native/mkldnn/UnaryOps.cpp", |
| 1058 | "aten/src/ATen/native/mkldnn/Utils.cpp", |
haozhe.zhu | 752a820 | 2021-09-17 09:52:47 -0700 | [diff] [blame] | 1059 | "aten/src/ATen/native/mkldnn/Matmul.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1060 | "aten/src/ATen/native/quantized/cpu/init_qnnpack.cpp", |
Kimish Patel | efe2c04 | 2022-07-16 06:02:01 +0000 | [diff] [blame] | 1061 | # This is moved to aten_cpu because some of the custom ops use empty_with_tail_padding |
| 1062 | # which was available only within aten_native_cpu. Ideally the right fix is to make |
| 1063 | # empty_with_tail_padding into an op and use dispatcher with it. But exposing it as an op |
| 1064 | # has limited use and hence does not seem to really make sense. |
| 1065 | "aten/src/ATen/native/utils/Factory.cpp", |
Victor Quach | 5abeac3 | 2021-08-13 07:47:12 -0700 | [diff] [blame] | 1066 | "aten/src/ATen/SavedTensorHooks.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1067 | "aten/src/ATen/vulkan/Context.cpp", |
Pavithran Ramachandran | 8beabff | 2021-10-20 20:40:12 -0700 | [diff] [blame] | 1068 | "aten/src/ATen/native/prim_native_functions.cpp", |
Jing Xu | 0e95746 | 2022-07-27 23:29:35 +0000 | [diff] [blame] | 1069 | "aten/src/ATen/native/verbose_wrapper.cpp", |
leslie-fang-intel | 9832cfb | 2023-06-19 14:52:39 +0800 | [diff] [blame] | 1070 | "aten/src/ATen/cpu/Utils.cpp", |
Nikita Shulga | 6cef594 | 2022-12-31 00:02:43 +0000 | [diff] [blame] | 1071 | ] + aten_cpu_non_globed_sources |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1072 | |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1073 | aten_cpu_source_codegen_list = [ |
| 1074 | "aten/src/ATen/native/cpu/AdaptiveAvgPoolKernel.cpp", |
mingfeima | dd045ab | 2021-06-25 16:34:57 -0700 | [diff] [blame] | 1075 | "aten/src/ATen/native/cpu/AdaptiveMaxPoolKernel.cpp", |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1076 | ] |
| 1077 | |
Edward Yang | ce7910b | 2022-02-28 15:46:04 -0800 | [diff] [blame] | 1078 | aten_ufunc_headers = [ |
| 1079 | "aten/src/ATen/native/ufunc/add.h", |
| 1080 | ] |
| 1081 | |
H1Gdev | d7d0fa2 | 2021-05-21 11:43:47 -0700 | [diff] [blame] | 1082 | # When building lite interpreter in OSS, "aten/src/ATen/native/cpu/AdaptiveAvgPoolKernel.cpp" will go through |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1083 | # codegen process. The codegen version of this file, like Activation.cpp.DEFAULT.cpp, will be included |
| 1084 | # in ${cpu_kernel_cpp} in aten/src/ATen/CMakeLists.txt. As a result, in aten/src/ATen/CMakeLists.txt, |
| 1085 | # only aten_cpu_source_non_codegen_list need to be added to ${all_cpu_cpp}. |
| 1086 | aten_cpu_source_list = sorted(aten_cpu_source_non_codegen_list + aten_cpu_source_codegen_list) |
| 1087 | |
| 1088 | # Same as ${aten_cpu_source_codegen_list}, this list will go through aten codegen, and be included in |
| 1089 | # ${cpu_kernel_cpp} in aten/src/ATen/CMakeLists.txt. |
| 1090 | aten_native_source_codegen_list = [ |
| 1091 | "aten/src/ATen/native/cpu/Activation.cpp", |
mingfeima | bb19dc1 | 2021-06-09 21:05:35 -0700 | [diff] [blame] | 1092 | "aten/src/ATen/native/cpu/AvgPoolKernel.cpp", |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1093 | "aten/src/ATen/native/cpu/BinaryOpsKernel.cpp", |
| 1094 | "aten/src/ATen/native/cpu/BlasKernel.cpp", |
| 1095 | "aten/src/ATen/native/cpu/CatKernel.cpp", |
mingfeima | 054b90f | 2022-01-14 11:54:06 -0800 | [diff] [blame] | 1096 | "aten/src/ATen/native/cpu/ChannelShuffleKernel.cpp", |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1097 | "aten/src/ATen/native/cpu/ComplexKernel.cpp", |
| 1098 | "aten/src/ATen/native/cpu/CopyKernel.cpp", |
| 1099 | "aten/src/ATen/native/cpu/CrossKernel.cpp", |
| 1100 | "aten/src/ATen/native/cpu/DepthwiseConvKernel.cpp", |
| 1101 | "aten/src/ATen/native/cpu/DistanceOpsKernel.cpp", |
Peter Bell | 1841f76 | 2021-10-12 08:09:29 -0700 | [diff] [blame] | 1102 | "aten/src/ATen/native/cpu/DistributionKernels.cpp", |
Liao, Xuan | 5516fe1 | 2023-08-18 21:22:08 +0000 | [diff] [blame] | 1103 | "aten/src/ATen/native/cpu/FlashAttentionKernel.cpp", |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1104 | "aten/src/ATen/native/cpu/FillKernel.cpp", |
| 1105 | "aten/src/ATen/native/cpu/FunctionOfAMatrixUtilsKernel.cpp", |
| 1106 | "aten/src/ATen/native/cpu/GridSamplerKernel.cpp", |
Saketh Are | 729f7cd | 2021-06-22 10:04:51 -0700 | [diff] [blame] | 1107 | "aten/src/ATen/native/cpu/HistogramKernel.cpp", |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1108 | "aten/src/ATen/native/cpu/IndexKernel.cpp", |
| 1109 | "aten/src/ATen/native/cpu/LerpKernel.cpp", |
| 1110 | "aten/src/ATen/native/cpu/LinearAlgebraKernel.cpp", |
mingfeima | 1e03a25 | 2021-04-20 15:00:55 -0700 | [diff] [blame] | 1111 | "aten/src/ATen/native/cpu/MaxPoolKernel.cpp", |
Allen Goodman | b3ca363 | 2022-06-24 20:57:38 +0000 | [diff] [blame] | 1112 | "aten/src/ATen/native/cpu/MaxPooling.cpp", |
mingfeima | c5ed31e | 2021-08-29 18:35:37 -0700 | [diff] [blame] | 1113 | "aten/src/ATen/native/cpu/MaxUnpoolKernel.cpp", |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1114 | "aten/src/ATen/native/cpu/MultinomialKernel.cpp", |
mingfeima | a683bc5 | 2023-09-18 09:49:50 +0800 | [diff] [blame] | 1115 | "aten/src/ATen/native/cpu/NativeMultiheadAttnKernel.cpp", |
mingfeima | 69b09ec | 2023-06-14 13:02:38 +0800 | [diff] [blame] | 1116 | "aten/src/ATen/native/cpu/PaddingKernel.cpp", |
mingfeima | c003494 | 2022-05-13 08:46:37 +0800 | [diff] [blame] | 1117 | "aten/src/ATen/native/cpu/PixelShuffleKernel.cpp", |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1118 | "aten/src/ATen/native/cpu/PointwiseOpsKernel.cpp", |
| 1119 | "aten/src/ATen/native/cpu/PowKernel.cpp", |
| 1120 | "aten/src/ATen/native/cpu/RangeFactoriesKernel.cpp", |
| 1121 | "aten/src/ATen/native/cpu/ReduceAllOpsKernel.cpp", |
| 1122 | "aten/src/ATen/native/cpu/ReduceOpsKernel.cpp", |
Peter Bell | 6408cbd | 2021-06-03 11:42:14 -0700 | [diff] [blame] | 1123 | "aten/src/ATen/native/cpu/RenormKernel.cpp", |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1124 | "aten/src/ATen/native/cpu/ScatterGatherKernel.cpp", |
| 1125 | "aten/src/ATen/native/cpu/SoftMaxKernel.cpp", |
| 1126 | "aten/src/ATen/native/cpu/SortingKernel.cpp", |
| 1127 | "aten/src/ATen/native/cpu/StackKernel.cpp", |
| 1128 | "aten/src/ATen/native/cpu/SumKernel.cpp", |
| 1129 | "aten/src/ATen/native/cpu/TensorCompareKernel.cpp", |
| 1130 | "aten/src/ATen/native/cpu/UnaryOpsKernel.cpp", |
| 1131 | "aten/src/ATen/native/cpu/Unfold2d.cpp", |
| 1132 | "aten/src/ATen/native/cpu/UnfoldBackwardKernel.cpp", |
| 1133 | "aten/src/ATen/native/cpu/UpSampleKernel.cpp", |
| 1134 | "aten/src/ATen/native/cpu/UpSampleMoreKernel.cpp", |
Allen Goodman | b3ca363 | 2022-06-24 20:57:38 +0000 | [diff] [blame] | 1135 | "aten/src/ATen/native/cpu/WeightNormKernel.cpp", |
Allen Goodman | b3308e2 | 2022-06-23 19:33:40 +0000 | [diff] [blame] | 1136 | "aten/src/ATen/native/cpu/airy_ai.cpp", |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1137 | "aten/src/ATen/native/cpu/batch_norm_kernel.cpp", |
| 1138 | "aten/src/ATen/native/cpu/group_norm_kernel.cpp", |
| 1139 | "aten/src/ATen/native/cpu/layer_norm_kernel.cpp", |
Allen Goodman | 63ef2a0 | 2022-06-29 14:53:37 +0000 | [diff] [blame] | 1140 | "aten/src/ATen/native/cpu/scaled_modified_bessel_k0.cpp", |
Allen Goodman | b3ca363 | 2022-06-24 20:57:38 +0000 | [diff] [blame] | 1141 | "aten/src/ATen/native/cpu/scaled_modified_bessel_k1.cpp", |
Allen Goodman | ab8797d | 2022-06-27 20:14:46 +0000 | [diff] [blame] | 1142 | "aten/src/ATen/native/cpu/spherical_bessel_j0.cpp", |
mingfeima | 3ab58fd | 2023-01-12 10:19:29 +0800 | [diff] [blame] | 1143 | "aten/src/ATen/native/cpu/SampledAddmmKernel.cpp", |
mingfeima | c620ece | 2023-02-10 11:12:35 +0800 | [diff] [blame] | 1144 | "aten/src/ATen/native/cpu/SpmmReduceKernel.cpp", |
Andrew M. James | 5a4c9e8 | 2022-06-29 16:50:32 -0500 | [diff] [blame] | 1145 | "aten/src/ATen/native/cpu/SparseFactories.cpp", |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1146 | "aten/src/ATen/native/quantized/cpu/kernels/QuantizedOpKernels.cpp", |
| 1147 | ] |
| 1148 | |
| 1149 | # This aten native source file list will not go through aten codegen process |
| 1150 | aten_native_source_non_codegen_list = [ |
Kimish Patel | db42ec4 | 2021-05-18 14:16:10 -0700 | [diff] [blame] | 1151 | "aten/src/ATen/native/ao_sparse/library.cpp", |
| 1152 | "aten/src/ATen/native/ao_sparse/quantized/cpu/fbgemm_utils.cpp", |
| 1153 | "aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear.cpp", |
Salil Desai | eaf817d | 2022-07-06 10:56:13 -0700 | [diff] [blame] | 1154 | "aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear_deserialize.cpp", |
Kimish Patel | db42ec4 | 2021-05-18 14:16:10 -0700 | [diff] [blame] | 1155 | "aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear_dynamic.cpp", |
| 1156 | "aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear_prepack.cpp", |
Salil Desai | eaf817d | 2022-07-06 10:56:13 -0700 | [diff] [blame] | 1157 | "aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear_serialize.cpp", |
Kimish Patel | db42ec4 | 2021-05-18 14:16:10 -0700 | [diff] [blame] | 1158 | "aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear_unpack.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1159 | "aten/src/ATen/native/quantized/cpu/fbgemm_utils.cpp", |
Supriya Rao | 92d3391 | 2021-07-21 10:01:19 -0700 | [diff] [blame] | 1160 | "aten/src/ATen/native/quantized/cpu/fused_obs_fake_quant.cpp", |
dzdang | 5874a31 | 2022-05-31 11:02:37 -0700 | [diff] [blame] | 1161 | "aten/src/ATen/native/quantized/cpu/IntReprQuant.cpp", |
| 1162 | "aten/src/ATen/native/quantized/cpu/MakePerTensorQuantizedTensor.cpp", |
| 1163 | "aten/src/ATen/native/quantized/cpu/AdaptiveAveragePooling.cpp", |
| 1164 | "aten/src/ATen/native/quantized/cpu/AveragePool2d.cpp", |
| 1165 | "aten/src/ATen/native/quantized/cpu/AveragePool3d.cpp", |
| 1166 | "aten/src/ATen/native/quantized/cpu/BinaryOps.cpp", |
| 1167 | "aten/src/ATen/native/quantized/cpu/Normalization.cpp", |
| 1168 | "aten/src/ATen/native/quantized/cpu/ChannelShuffle.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1169 | "aten/src/ATen/native/quantized/cpu/qclamp.cpp", |
dzdang | 5874a31 | 2022-05-31 11:02:37 -0700 | [diff] [blame] | 1170 | "aten/src/ATen/native/quantized/cpu/TensorShape.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1171 | "aten/src/ATen/native/quantized/cpu/qconv.cpp", |
| 1172 | "aten/src/ATen/native/quantized/cpu/qconv_prepack.cpp", |
David Dang | 3a261f8 | 2022-03-17 06:01:57 -0700 | [diff] [blame] | 1173 | "aten/src/ATen/native/quantized/cpu/qconv_unpack_impl.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1174 | "aten/src/ATen/native/quantized/cpu/qelu.cpp", |
| 1175 | "aten/src/ATen/native/quantized/cpu/qembeddingbag.cpp", |
| 1176 | "aten/src/ATen/native/quantized/cpu/qembeddingbag_prepack.cpp", |
| 1177 | "aten/src/ATen/native/quantized/cpu/qembeddingbag_unpack.cpp", |
Weiwen Xia | 32bf5e0 | 2022-01-06 18:59:49 -0800 | [diff] [blame] | 1178 | "aten/src/ATen/native/quantized/cpu/qgelu.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1179 | "aten/src/ATen/native/quantized/cpu/qhardsigmoid.cpp", |
| 1180 | "aten/src/ATen/native/quantized/cpu/qhardswish.cpp", |
| 1181 | "aten/src/ATen/native/quantized/cpu/qlinear.cpp", |
| 1182 | "aten/src/ATen/native/quantized/cpu/qlinear_dynamic.cpp", |
Charles David Hernandez | 09615cd | 2021-11-15 16:38:35 -0800 | [diff] [blame] | 1183 | "aten/src/ATen/native/quantized/cpu/qconv_dynamic.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1184 | "aten/src/ATen/native/quantized/cpu/qlinear_prepack.cpp", |
dzdang | 5874a31 | 2022-05-31 11:02:37 -0700 | [diff] [blame] | 1185 | "aten/src/ATen/native/quantized/cpu/LinearUnpackImpl.cpp", |
Salil Desai | 26b8a00 | 2022-02-04 17:36:40 -0800 | [diff] [blame] | 1186 | "aten/src/ATen/native/quantized/cpu/qmatmul.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1187 | "aten/src/ATen/native/quantized/cpu/qmul.cpp", |
| 1188 | "aten/src/ATen/native/quantized/cpu/qnormalization.cpp", |
dzdang | 5874a31 | 2022-05-31 11:02:37 -0700 | [diff] [blame] | 1189 | "aten/src/ATen/native/quantized/cpu/Pooling.cpp", |
| 1190 | "aten/src/ATen/native/quantized/cpu/ReduceOps.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1191 | "aten/src/ATen/native/quantized/cpu/qrelu.cpp", |
| 1192 | "aten/src/ATen/native/quantized/cpu/qsigmoid.cpp", |
Salil Desai | ca0ef52 | 2022-04-11 22:39:50 +0000 | [diff] [blame] | 1193 | "aten/src/ATen/native/quantized/cpu/qsoftmax.cpp", |
dzdang | 5874a31 | 2022-05-31 11:02:37 -0700 | [diff] [blame] | 1194 | "aten/src/ATen/native/quantized/cpu/Sorting.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1195 | "aten/src/ATen/native/quantized/cpu/qtanh.cpp", |
| 1196 | "aten/src/ATen/native/quantized/cpu/qthreshold.cpp", |
dzdang | 5874a31 | 2022-05-31 11:02:37 -0700 | [diff] [blame] | 1197 | "aten/src/ATen/native/quantized/cpu/UpSampleBilinear2d.cpp", |
| 1198 | "aten/src/ATen/native/quantized/cpu/UpSampleNearest2d.cpp", |
| 1199 | "aten/src/ATen/native/quantized/cpu/UpSampleNearest3d.cpp", |
| 1200 | "aten/src/ATen/native/quantized/cpu/TensorOperators.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1201 | "aten/src/ATen/native/quantized/Copy.cpp", |
| 1202 | "aten/src/ATen/native/quantized/QTensor.cpp", |
| 1203 | "aten/src/ATen/native/quantized/TensorCompare.cpp", |
| 1204 | "aten/src/ATen/native/quantized/TensorFactories.cpp", |
dzdang | a56f4e2 | 2022-06-06 10:49:37 -0400 | [diff] [blame] | 1205 | "aten/src/ATen/native/quantized/AffineQuantizer.cpp", |
| 1206 | "aten/src/ATen/native/quantized/AffineQuantizerBase.cpp", |
| 1207 | "aten/src/ATen/native/quantized/FakeQuantPerChannelAffine.cpp", |
| 1208 | "aten/src/ATen/native/quantized/FakeQuantPerTensorAffine.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1209 | "aten/src/ATen/native/quantized/library.cpp", |
dzdang | 2b6a042 | 2022-06-08 08:18:34 -0700 | [diff] [blame] | 1210 | "aten/src/ATen/native/quantized/TensorAdvancedIndexing.cpp", |
dzdang | 5874a31 | 2022-05-31 11:02:37 -0700 | [diff] [blame] | 1211 | "aten/src/ATen/native/quantized/cpu/RuyUtils.cpp", |
| 1212 | "aten/src/ATen/native/quantized/cpu/XnnpackUtils.cpp", |
Mengwei Liu | e13229e | 2022-03-22 15:14:33 -0700 | [diff] [blame] | 1213 | "aten/src/ATen/native/quantized/qlinear_unpack.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1214 | "aten/src/ATen/quantized/QTensorImpl.cpp", |
| 1215 | "aten/src/ATen/quantized/Quantizer.cpp", |
| 1216 | "aten/src/ATen/native/Activation.cpp", |
| 1217 | "aten/src/ATen/native/AdaptiveAveragePooling.cpp", |
| 1218 | "aten/src/ATen/native/AdaptiveAveragePooling3d.cpp", |
| 1219 | "aten/src/ATen/native/AdaptiveMaxPooling2d.cpp", |
| 1220 | "aten/src/ATen/native/AdaptiveMaxPooling3d.cpp", |
| 1221 | "aten/src/ATen/native/AffineGridGenerator.cpp", |
| 1222 | "aten/src/ATen/native/AveragePool2d.cpp", |
| 1223 | "aten/src/ATen/native/AveragePool3d.cpp", |
| 1224 | "aten/src/ATen/native/BatchLinearAlgebra.cpp", |
Nikita Shulga | d0dcebe | 2022-05-06 18:48:42 -0700 | [diff] [blame] | 1225 | "aten/src/ATen/native/BatchLinearAlgebraKernel.cpp", |
Richard Zou | 4b10534 | 2022-12-07 07:48:41 -0800 | [diff] [blame] | 1226 | "aten/src/ATen/native/LegacyBatching.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1227 | "aten/src/ATen/native/BinaryOps.cpp", |
| 1228 | "aten/src/ATen/native/Blas.cpp", |
| 1229 | "aten/src/ATen/native/BlasKernel.cpp", |
| 1230 | "aten/src/ATen/native/Bucketization.cpp", |
| 1231 | "aten/src/ATen/native/CPUBlas.cpp", |
| 1232 | "aten/src/ATen/native/ChanelShuffle.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1233 | "aten/src/ATen/native/Col2Im.cpp", |
Peter Bell | f86bb2d | 2022-03-19 12:13:43 +0000 | [diff] [blame] | 1234 | "aten/src/ATen/native/PadNd.cpp", |
xuanqi | b27c355 | 2023-06-15 16:27:30 -0700 | [diff] [blame] | 1235 | "aten/src/ATen/native/Constraints.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1236 | "aten/src/ATen/native/Convolution.cpp", |
| 1237 | "aten/src/ATen/native/ConvolutionMM2d.cpp", |
| 1238 | "aten/src/ATen/native/ConvolutionMM3d.cpp", |
| 1239 | "aten/src/ATen/native/ConvolutionTBC.cpp", |
| 1240 | "aten/src/ATen/native/Copy.cpp", |
Heitor Schueroff | ec9c03c | 2021-06-29 13:59:46 -0700 | [diff] [blame] | 1241 | "aten/src/ATen/native/Correlation.cpp", |
Brian Hirsh | 9134b0e | 2021-06-25 16:25:31 -0700 | [diff] [blame] | 1242 | "aten/src/ATen/native/CPUFallback.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1243 | "aten/src/ATen/native/Cross.cpp", |
| 1244 | "aten/src/ATen/native/DilatedMaxPool2d.cpp", |
| 1245 | "aten/src/ATen/native/DilatedMaxPool3d.cpp", |
| 1246 | # Referenced by both native and ATen/Version.cpp. Does not reference to other native symbols |
| 1247 | # "aten/src/ATen/native/DispatchStub.cpp", |
| 1248 | # "aten/src/ATen/native/quantized/cpu/init_qnnpack.cpp", |
| 1249 | "aten/src/ATen/native/Distance.cpp", |
| 1250 | "aten/src/ATen/native/Distributions.cpp", |
| 1251 | "aten/src/ATen/native/Dropout.cpp", |
| 1252 | "aten/src/ATen/native/Embedding.cpp", |
| 1253 | "aten/src/ATen/native/EmbeddingBag.cpp", |
| 1254 | "aten/src/ATen/native/Fill.cpp", |
| 1255 | "aten/src/ATen/native/ForeachOpsKernels.cpp", |
| 1256 | "aten/src/ATen/native/FractionalMaxPool2d.cpp", |
| 1257 | "aten/src/ATen/native/FractionalMaxPool3d.cpp", |
| 1258 | "aten/src/ATen/native/FunctionOfAMatrixUtils.cpp", |
| 1259 | "aten/src/ATen/native/GatedLinearUnit.cpp", |
| 1260 | "aten/src/ATen/native/GridSampler.cpp", |
Saketh Are | 729f7cd | 2021-06-22 10:04:51 -0700 | [diff] [blame] | 1261 | "aten/src/ATen/native/Histogram.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1262 | "aten/src/ATen/native/Im2Col.cpp", |
| 1263 | "aten/src/ATen/native/IndexingUtils.cpp", |
| 1264 | "aten/src/ATen/native/Integration.cpp", |
| 1265 | "aten/src/ATen/native/Itertools.cpp", |
| 1266 | "aten/src/ATen/native/LegacyBridge.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1267 | "aten/src/ATen/native/Lerp.cpp", |
| 1268 | "aten/src/ATen/native/Linear.cpp", |
| 1269 | "aten/src/ATen/native/LinearAlgebra.cpp", |
| 1270 | "aten/src/ATen/native/Loss.cpp", |
| 1271 | "aten/src/ATen/native/LossCTC.cpp", |
| 1272 | "aten/src/ATen/native/LossMultiLabelMargin.cpp", |
| 1273 | "aten/src/ATen/native/LossMultiMargin.cpp", |
| 1274 | "aten/src/ATen/native/LossNLL.cpp", |
| 1275 | "aten/src/ATen/native/LossNLL2d.cpp", |
| 1276 | "aten/src/ATen/native/MaxPooling.cpp", |
| 1277 | "aten/src/ATen/native/MaxUnpooling.cpp", |
| 1278 | "aten/src/ATen/native/Memory.cpp", |
| 1279 | "aten/src/ATen/native/MetaTensor.cpp", |
| 1280 | "aten/src/ATen/native/NNPACK.cpp", |
| 1281 | "aten/src/ATen/native/NaiveConvolutionTranspose2d.cpp", |
| 1282 | "aten/src/ATen/native/NaiveConvolutionTranspose3d.cpp", |
| 1283 | "aten/src/ATen/native/NaiveDilatedConvolution.cpp", |
| 1284 | "aten/src/ATen/native/NamedTensor.cpp", |
| 1285 | "aten/src/ATen/native/Normalization.cpp", |
| 1286 | "aten/src/ATen/native/Onehot.cpp", |
| 1287 | "aten/src/ATen/native/PackedSequence.cpp", |
| 1288 | "aten/src/ATen/native/PixelShuffle.cpp", |
| 1289 | "aten/src/ATen/native/PointwiseOps.cpp", |
| 1290 | "aten/src/ATen/native/Pooling.cpp", |
| 1291 | "aten/src/ATen/native/Pow.cpp", |
PyTorch MergeBot | 5ce8002 | 2023-10-25 16:34:16 +0000 | [diff] [blame] | 1292 | "aten/src/ATen/native/QuantizedLinear.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1293 | "aten/src/ATen/native/RNN.cpp", |
| 1294 | "aten/src/ATen/native/RangeFactories.cpp", |
| 1295 | "aten/src/ATen/native/ReduceAllOps.cpp", |
| 1296 | "aten/src/ATen/native/ReduceOps.cpp", |
| 1297 | "aten/src/ATen/native/ReflectionPad.cpp", |
| 1298 | "aten/src/ATen/native/Repeat.cpp", |
| 1299 | "aten/src/ATen/native/ReplicationPadding.cpp", |
| 1300 | "aten/src/ATen/native/Resize.cpp", |
| 1301 | "aten/src/ATen/native/RowwisePrune.cpp", |
Serhat Yilmaz | 7e3cf1e | 2021-03-23 15:56:00 -0700 | [diff] [blame] | 1302 | "aten/src/ATen/native/SegmentReduce.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1303 | "aten/src/ATen/native/Scalar.cpp", |
| 1304 | "aten/src/ATen/native/SobolEngineOps.cpp", |
| 1305 | "aten/src/ATen/native/SobolEngineOpsUtils.cpp", |
| 1306 | "aten/src/ATen/native/SoftMax.cpp", |
| 1307 | "aten/src/ATen/native/Sorting.cpp", |
Nikita Shulga | 82daf98 | 2023-03-14 02:56:52 +0000 | [diff] [blame] | 1308 | "aten/src/ATen/native/SparseTensorUtils.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1309 | "aten/src/ATen/native/SpectralOps.cpp", |
| 1310 | "aten/src/ATen/native/SummaryOps.cpp", |
| 1311 | "aten/src/ATen/native/TensorAdvancedIndexing.cpp", |
| 1312 | "aten/src/ATen/native/TensorCompare.cpp", |
| 1313 | "aten/src/ATen/native/TensorConversions.cpp", |
| 1314 | "aten/src/ATen/native/TensorFactories.cpp", |
| 1315 | "aten/src/ATen/native/TensorIteratorReduce.cpp", |
| 1316 | "aten/src/ATen/native/TensorProperties.cpp", |
| 1317 | "aten/src/ATen/native/TensorShape.cpp", |
| 1318 | "aten/src/ATen/native/TensorTransformations.cpp", |
| 1319 | "aten/src/ATen/native/TestOps.cpp", |
| 1320 | "aten/src/ATen/native/TriangularOps.cpp", |
| 1321 | "aten/src/ATen/native/TypeProperties.cpp", |
| 1322 | "aten/src/ATen/native/UnaryOps.cpp", |
| 1323 | "aten/src/ATen/native/Unfold2d.cpp", |
| 1324 | "aten/src/ATen/native/Unfold3d.cpp", |
| 1325 | "aten/src/ATen/native/UnfoldBackward.cpp", |
| 1326 | "aten/src/ATen/native/Unique.cpp", |
| 1327 | # Low-level functions that can be directly referenced |
| 1328 | # "aten/src/ATen/native/UpSample.cpp", |
| 1329 | "aten/src/ATen/native/UpSampleBicubic2d.cpp", |
| 1330 | "aten/src/ATen/native/UpSampleBilinear2d.cpp", |
| 1331 | "aten/src/ATen/native/UpSampleLinear1d.cpp", |
| 1332 | "aten/src/ATen/native/UpSampleNearest1d.cpp", |
| 1333 | "aten/src/ATen/native/UpSampleNearest2d.cpp", |
| 1334 | "aten/src/ATen/native/UpSampleNearest3d.cpp", |
| 1335 | "aten/src/ATen/native/UpSampleTrilinear3d.cpp", |
| 1336 | "aten/src/ATen/native/VariableMethodStubs.cpp", |
| 1337 | "aten/src/ATen/native/WeightNorm.cpp", |
| 1338 | "aten/src/ATen/native/group_norm.cpp", |
| 1339 | "aten/src/ATen/native/layer_norm.cpp", |
Max Ren | 0b3a239 | 2022-07-21 02:45:20 +0000 | [diff] [blame] | 1340 | "aten/src/ATen/native/mkl/LinearAlgebra.cpp", |
| 1341 | "aten/src/ATen/native/mkl/SparseBlasImpl.cpp", |
| 1342 | "aten/src/ATen/native/mkl/SparseCsrLinearAlgebra.cpp", |
| 1343 | "aten/src/ATen/native/mkl/SpectralOps.cpp", |
Christian Puhrsch | 6fe41e7 | 2022-10-31 20:10:05 +0000 | [diff] [blame] | 1344 | "aten/src/ATen/native/nested/NestedTensorAliases.cpp", |
| 1345 | "aten/src/ATen/native/nested/NestedTensorBackward.cpp", |
| 1346 | "aten/src/ATen/native/nested/NestedTensorBinaryOps.cpp", |
Antoni Viros i Martin | d94e33f | 2022-10-20 03:46:48 +0000 | [diff] [blame] | 1347 | "aten/src/ATen/native/nested/NestedTensorFactories.cpp", |
Christian Puhrsch | 484c0de | 2022-03-02 07:29:19 -0800 | [diff] [blame] | 1348 | "aten/src/ATen/native/nested/NestedTensorMath.cpp", |
Christian Puhrsch | 6fe41e7 | 2022-10-31 20:10:05 +0000 | [diff] [blame] | 1349 | "aten/src/ATen/native/nested/NestedTensorMatmul.cpp", |
Scott Wolchok | 97c993c | 2022-04-13 09:46:33 -0700 | [diff] [blame] | 1350 | "aten/src/ATen/native/nested/NestedTensorTransformerFunctions.cpp", |
Christian Puhrsch | 6fe41e7 | 2022-10-31 20:10:05 +0000 | [diff] [blame] | 1351 | "aten/src/ATen/native/nested/NestedTensorUnaryOps.cpp", |
Driss Guessous | f803fa9 | 2022-09-02 16:31:55 +0000 | [diff] [blame] | 1352 | "aten/src/ATen/native/nested/NestedTensorUtils.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1353 | "aten/src/ATen/native/sparse/ParamUtils.cpp", |
| 1354 | "aten/src/ATen/native/sparse/SoftMax.cpp", |
Ivan Yashchuk | cbf596b | 2021-11-09 12:30:24 -0800 | [diff] [blame] | 1355 | "aten/src/ATen/native/sparse/SparseBlas.cpp", |
| 1356 | "aten/src/ATen/native/sparse/SparseBlasImpl.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1357 | "aten/src/ATen/native/sparse/SparseMatMul.cpp", |
| 1358 | "aten/src/ATen/native/sparse/SparseTensor.cpp", |
Sameer Deshmukh | 5fb1142 | 2021-04-12 10:07:56 -0700 | [diff] [blame] | 1359 | "aten/src/ATen/native/sparse/SparseCsrTensor.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1360 | "aten/src/ATen/native/sparse/SparseTensorMath.cpp", |
Peter Bell | 1da1707 | 2021-12-01 05:41:43 -0800 | [diff] [blame] | 1361 | "aten/src/ATen/native/sparse/SparseUnaryOps.cpp", |
Sameer Deshmukh | 5fb1142 | 2021-04-12 10:07:56 -0700 | [diff] [blame] | 1362 | "aten/src/ATen/native/sparse/SparseCsrTensorMath.cpp", |
Andrew M. James | 5a4c9e8 | 2022-06-29 16:50:32 -0500 | [diff] [blame] | 1363 | "aten/src/ATen/native/sparse/SparseFactories.cpp", |
Nikita Vedeneev | 880b972 | 2022-07-14 20:36:18 +0000 | [diff] [blame] | 1364 | "aten/src/ATen/native/sparse/ValidateCompressedIndicesKernel.cpp", |
nikitaved | 12ae3be | 2022-09-23 23:31:17 +0000 | [diff] [blame] | 1365 | "aten/src/ATen/native/sparse/SparseBinaryOpIntersectionKernel.cpp", |
Nikita Vedeneev | d0f4d62 | 2023-03-09 11:13:43 +0000 | [diff] [blame] | 1366 | "aten/src/ATen/native/sparse/FlattenIndicesKernel.cpp", |
Scott Wolchok | e816e17 | 2022-04-25 17:02:39 -0700 | [diff] [blame] | 1367 | "aten/src/ATen/native/transformers/attention.cpp", |
Liao, Xuan | 71632d4 | 2023-08-19 16:14:37 +0000 | [diff] [blame] | 1368 | "aten/src/ATen/native/transformers/sdp_utils_cpp.cpp", |
Scott Wolchok | e816e17 | 2022-04-25 17:02:39 -0700 | [diff] [blame] | 1369 | "aten/src/ATen/native/transformers/transformer.cpp", |
Akshit Khurana | 28fc59d | 2021-04-28 14:52:37 -0700 | [diff] [blame] | 1370 | "aten/src/ATen/native/xnnpack/Activation.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1371 | "aten/src/ATen/native/xnnpack/ChannelShuffle.cpp", |
| 1372 | "aten/src/ATen/native/xnnpack/Convolution.cpp", |
Akshit Khurana | 3483049 | 2021-04-28 14:52:37 -0700 | [diff] [blame] | 1373 | "aten/src/ATen/native/xnnpack/AveragePooling.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1374 | "aten/src/ATen/native/xnnpack/Init.cpp", |
| 1375 | "aten/src/ATen/native/xnnpack/Linear.cpp", |
| 1376 | "aten/src/ATen/native/xnnpack/MaxPooling.cpp", |
| 1377 | "aten/src/ATen/native/xnnpack/OpContext.cpp", |
| 1378 | "aten/src/ATen/native/xnnpack/RegisterOpContextClass.cpp", |
| 1379 | "aten/src/ATen/native/xnnpack/Shim.cpp", |
| 1380 | # Files not in native, but depends on native symbols |
| 1381 | # "aten/src/ATen/TensorIndexing.cpp", |
| 1382 | "aten/src/ATen/TensorIterator.cpp", |
Chen Lai | 502a859 | 2021-02-26 11:24:10 -0800 | [diff] [blame] | 1383 | ] |
Chen Lai | 98943bb | 2021-03-08 12:23:00 -0800 | [diff] [blame] | 1384 | |
| 1385 | # 1. Files in ATen/native with a few exceptions |
| 1386 | # TODO: move the exceptions to proper locations |
| 1387 | # 2. The whole aten native source list includes the list with and without aten codegen process. |
| 1388 | aten_native_source_list = sorted(aten_native_source_non_codegen_list + aten_native_source_codegen_list) |
Peter Bell | bb456d2 | 2022-01-31 18:40:49 -0800 | [diff] [blame] | 1389 | |
| 1390 | # These are cpp files which need to go in the torch_cuda_cu library |
| 1391 | # .cu files can be found via glob |
| 1392 | aten_cuda_cu_source_list = [ |
| 1393 | "aten/src/ATen/cuda/CUDABlas.cpp", |
Peter Bell | bb456d2 | 2022-01-31 18:40:49 -0800 | [diff] [blame] | 1394 | "aten/src/ATen/cuda/CUDASparseBlas.cpp", |
| 1395 | "aten/src/ATen/cuda/CublasHandlePool.cpp", |
Peter Bell | bb456d2 | 2022-01-31 18:40:49 -0800 | [diff] [blame] | 1396 | "aten/src/ATen/native/cuda/Activation.cpp", |
Nikita Shulga | 6302cdb | 2022-03-01 12:58:49 -0800 | [diff] [blame] | 1397 | "aten/src/ATen/native/cuda/LinearAlgebraStubs.cpp", |
Peter Bell | bb456d2 | 2022-01-31 18:40:49 -0800 | [diff] [blame] | 1398 | "aten/src/ATen/native/cuda/Blas.cpp", |
Peter Bell | 1ac7bb0 | 2022-02-28 21:36:35 -0800 | [diff] [blame] | 1399 | "aten/src/ATen/native/cuda/Distributions.cpp", |
Peter Bell | bb456d2 | 2022-01-31 18:40:49 -0800 | [diff] [blame] | 1400 | "aten/src/ATen/native/cuda/Equal.cpp", |
Peter Bell | 28388b4 | 2022-02-15 07:11:37 -0800 | [diff] [blame] | 1401 | "aten/src/ATen/native/cuda/GridSampler.cpp", |
Peter Bell | bb456d2 | 2022-01-31 18:40:49 -0800 | [diff] [blame] | 1402 | "aten/src/ATen/native/cuda/IndexKernel.cpp", |
| 1403 | "aten/src/ATen/native/cuda/ReduceOps.cpp", |
| 1404 | "aten/src/ATen/native/cuda/ScanKernels.cpp", |
| 1405 | "aten/src/ATen/native/cuda/Sort.cpp", |
| 1406 | "aten/src/ATen/native/cuda/Sorting.cpp", |
| 1407 | "aten/src/ATen/native/cuda/TensorModeKernel.cpp", |
| 1408 | "aten/src/ATen/native/cuda/TensorShapeCUDA.cpp", |
| 1409 | "aten/src/ATen/native/cuda/TensorTopK.cpp", |
| 1410 | "aten/src/ATen/native/cuda/jit_utils.cpp", |
Scott Wolchok | 97c993c | 2022-04-13 09:46:33 -0700 | [diff] [blame] | 1411 | "aten/src/ATen/native/nested/cuda/NestedTensorTransformerFunctions.cpp", |
Peter Bell | bb456d2 | 2022-01-31 18:40:49 -0800 | [diff] [blame] | 1412 | "aten/src/ATen/native/sparse/cuda/SparseBlas.cpp", |
| 1413 | "aten/src/ATen/native/sparse/cuda/SparseBlasImpl.cpp", |
| 1414 | "aten/src/ATen/native/sparse/cuda/SparseBlasLegacy.cpp", |
| 1415 | "aten/src/ATen/native/sparse/cuda/SparseCUDABlas.cpp", |
drisspg | ad90ab3 | 2023-09-13 13:59:05 +0000 | [diff] [blame] | 1416 | "aten/src/ATen/native/transformers/cuda/flash_attn/flash_api.cpp", |
Peter Bell | bb456d2 | 2022-01-31 18:40:49 -0800 | [diff] [blame] | 1417 | ] |
| 1418 | |
| 1419 | # Files using thrust::sort_by_key need to be linked last |
| 1420 | aten_cuda_with_sort_by_key_source_list = [ |
| 1421 | # empty_cuda is needed by torch_cuda_cpp |
| 1422 | "aten/src/ATen/native/cuda/TensorFactories.cu", |
| 1423 | ] |
| 1424 | |
| 1425 | aten_cuda_cu_with_sort_by_key_source_list = [ |
| 1426 | "aten/src/ATen/native/cuda/Unique.cu", |
| 1427 | ] |
Chen Lai | bca75fd | 2022-11-12 21:41:31 -0800 | [diff] [blame] | 1428 | |
| 1429 | # Followings are source code for xnnpack delegate |
| 1430 | |
| 1431 | xnnpack_delegate_serializer_header = [ |
| 1432 | "torch/csrc/jit/backends/xnnpack/serialization/serializer.h", |
| 1433 | ] |
| 1434 | |
| 1435 | xnnpack_delegate_serializer_source_list = [ |
| 1436 | "torch/csrc/jit/backends/xnnpack/serialization/serializer.cpp", |
| 1437 | ] |
| 1438 | |
| 1439 | xnnpack_delegate_core_source_list = [ |
| 1440 | "torch/csrc/jit/backends/xnnpack/compiler/xnn_compiler.cpp", |
| 1441 | ] |
| 1442 | |
| 1443 | xnnpack_delegate_core_header = [ |
| 1444 | "torch/csrc/jit/backends/xnnpack/compiler/xnn_compiler.h", |
| 1445 | "torch/csrc/jit/backends/xnnpack/executor/xnn_executor.h", |
| 1446 | ] |
| 1447 | |
| 1448 | xnnpack_backend_header = [ |
| 1449 | "torch/csrc/jit/backends/xnnpack/xnnpack_graph_builder.h", |
| 1450 | ] + xnnpack_delegate_core_header |
| 1451 | |
| 1452 | xnnpack_backend_source_list = [ |
| 1453 | "torch/csrc/jit/backends/xnnpack/compiler/xnn_compiler.cpp", |
| 1454 | "torch/csrc/jit/backends/xnnpack/xnnpack_backend_lib.cpp", |
| 1455 | "torch/csrc/jit/backends/xnnpack/xnnpack_backend_preprocess.cpp", |
| 1456 | "torch/csrc/jit/backends/xnnpack/xnnpack_graph_builder.cpp", |
| 1457 | ] + xnnpack_delegate_core_source_list |