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 |