1 #pragma once 2 3 #include <c10/util/Exception.h> 4 5 #define INTRA_OP_PARALLEL 6 7 namespace at::internal { 8 9 TORCH_API void invoke_parallel( 10 const int64_t begin, 11 const int64_t end, 12 const int64_t grain_size, 13 const std::function<void(int64_t, int64_t)>& f); 14 15 } // namespace at::internal 16