venv/lib/python3.11/site-packages/torch/include/ATen/ParallelNative.h

#pragma once

#include <c10/util/Exception.h>

#define INTRA_OP_PARALLEL

namespace at::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 at::internal