1 #pragma once 2 3 #include <ATen/core/ivalue.h> 4 #include <c10/macros/Export.h> 5 #include <torch/csrc/inductor/aoti_torch/c/shim.h> 6 7 namespace torch::aot_inductor { 8 9 class ProxyExecutor { 10 public: 11 ProxyExecutor() = default; 12 virtual ~ProxyExecutor() = default; 13 14 virtual void call_function( 15 int extern_node_index, 16 int num_ints, 17 int64_t* flatten_int_args, 18 int num_tensors, 19 AtenTensorHandle* flatten_tensor_args) = 0; 20 }; 21 22 } // namespace torch::aot_inductor 23