mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
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>
9 lines
171 B
C++
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
|