pytorch/torch/csrc/cuda/python_comm.h
cyyever 456c87c8a2 [8/N] Fix extra warnings brought by clang-tidy-17 (#139151)
Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/139151
Approved by: https://github.com/ezyang

Co-authored-by: Aaron Gokaslan <aaronGokaslan@gmail.com>
2024-10-30 14:20:08 +00:00

9 lines
171 B
C++

#pragma once
#include <torch/csrc/utils/pythoncapi_compat.h>
namespace torch::cuda::python {
void initCommMethods(PyObject* module);
} // namespace torch::cuda::python