1 #pragma once 2 3 #include <ATen/core/ivalue.h> 4 #include <c10/macros/Macros.h> 5 #include <functional> 6 7 namespace at { 8 9 // Launches intra-op parallel task, returns a future 10 TORCH_API c10::intrusive_ptr<c10::ivalue::Future> intraop_launch_future( 11 std::function<void()> func); 12 13 } // namespace at 14