mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
6647475bc2
Previously, we the Variable.data PyObject* in THPVariable_Wrap. For many Variables, we don't access their data directly. Instead, they are passed from one Variable compuatation to another. This reduces the overhead of ATen-implemented Variable methods by ~200ns.
37 lines
1.1 KiB
C++
37 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include <Python.h>
|
|
#include <memory>
|
|
#include <ATen/ATen.h>
|
|
|
|
#include "torch/csrc/autograd/variable.h"
|
|
|
|
// Python object that backs torch.autograd.Variable
|
|
struct THPVariable {
|
|
PyObject_HEAD
|
|
// Payload
|
|
torch::autograd::Variable cdata;
|
|
// Tensor this wraps (corresponds to Python attr 'data').
|
|
// It assumed that a THPVariable is *uniquely* identified by the
|
|
// tensor it wraps.
|
|
// Invariant: v->data == v->cdata->data
|
|
PyObject* data;
|
|
// Hooks to be run on backwards pass (corresponds to Python attr
|
|
// '_backwards_hooks', set by 'register_hook')
|
|
PyObject* backward_hooks;
|
|
};
|
|
|
|
extern PyObject *THPVariableClass;
|
|
|
|
bool THPVariable_initModule(PyObject *module);
|
|
PyObject * THPVariable_NewVolatile(PyObject *data);
|
|
PyObject * THPVariable_NewLeaf(PyObject *data);
|
|
PyObject * THPVariable_NewWithFunction(PyObject *data, const std::shared_ptr<torch::autograd::Function>& var);
|
|
PyObject * THPVariable_Wrap(torch::autograd::Variable var);
|
|
PyObject * THPVariable_get_data(THPVariable *self);
|
|
|
|
inline bool THPVariable_Check(PyObject *obj)
|
|
{
|
|
return THPVariableClass && PyObject_IsInstance(obj, THPVariableClass);
|
|
}
|