mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
40d1f77384
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/68693
Generation of python bindings for native functions is split over 8
different files. One for each namespace, with the torch namespace
split into 3 shards, and methods in their own file as well. This
change ensures that editing any single (non-method) operator only
causes one of these files to be rebuilt.
Test Plan: Imported from OSS
Reviewed By: jbschlosser
Differential Revision: D32596270
Pulled By: albanD
fbshipit-source-id: 0570ec69e7476b8f1bc21138ba18fe8f95ebbe3f
(cherry picked from commit ba0fc71a3a)
31 lines
974 B
C
31 lines
974 B
C
#pragma once
|
|
|
|
#include <torch/csrc/Export.h>
|
|
#include <torch/csrc/python_headers.h>
|
|
#include <ATen/core/Generator.h>
|
|
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
|
|
struct THPGenerator {
|
|
PyObject_HEAD
|
|
at::Generator cdata;
|
|
};
|
|
|
|
// Creates a new Python object wrapping the default at::Generator. The reference is
|
|
// borrowed. The caller should ensure that the at::Generator object lifetime
|
|
// last at least as long as the Python wrapper.
|
|
TORCH_PYTHON_API PyObject * THPGenerator_initDefaultGenerator(at::Generator cdata);
|
|
|
|
#define THPGenerator_Check(obj) \
|
|
PyObject_IsInstance(obj, THPGeneratorClass)
|
|
|
|
TORCH_PYTHON_API extern PyObject *THPGeneratorClass;
|
|
|
|
bool THPGenerator_init(PyObject *module);
|
|
|
|
TORCH_PYTHON_API PyObject * THPGenerator_Wrap(at::Generator gen);
|
|
|
|
// Creates a new Python object for a Generator. The Generator must not already
|
|
// have a PyObject* associated with it.
|
|
PyObject* THPGenerator_NewWithVar(PyTypeObject* type, at::Generator gen);
|