Vehicle-cpp/hkpc/software/torch/include/ATen/ParallelNative.h

#pragma once

#include <algorithm>
#include <cstddef>
#include <exception>

#include <c10/util/Exception.h>

#define INTRA_OP_PARALLEL

namespace at {
namespace internal {

TORCH_API void invoke_parallel(
    const int64_t begin,
    const int64_t end,
    const int64_t grain_size,
    const std::function<void(int64_t, int64_t)>& f);

} // namespace internal

} // namespace at