mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
d05a11337c
Move functorch/functorch into `functorch` folder - Add functorch/CMakeLists.txt that adds `functorch` native python exension - Modify `setup.py` to package pytorch and functorch together into a single wheel - Modify `functorch.__version__` is not equal to that of `torch.__version__` - Add dummy `functorch/setup.py` file for the projects that still want to build it Differential Revision: [D39058811](https://our.internmc.facebook.com/intern/diff/D39058811) Pull Request resolved: https://github.com/pytorch/pytorch/pull/83464 Approved by: https://github.com/zou3519
49 lines
1.5 KiB
C
49 lines
1.5 KiB
C
// Copyright (c) Facebook, Inc. and its affiliates.
|
|
// All rights reserved.
|
|
//
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree.
|
|
|
|
#pragma once
|
|
// note: pytorch's python variable simple includes pybind which conflicts with minpybind
|
|
// so this file just reproduces the minimial API needed to extract Tensors from python objects.
|
|
|
|
#include <torch/csrc/python_headers.h>
|
|
#include <ATen/core/Tensor.h>
|
|
#include <torch/csrc/Export.h>
|
|
|
|
// Python object that backs torch.autograd.Variable
|
|
// NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
|
|
struct THPVariable {
|
|
PyObject_HEAD;
|
|
// Payload
|
|
c10::MaybeOwned<at::Tensor> cdata;
|
|
// Hooks to be run on backwards pass (corresponds to Python attr
|
|
// '_backwards_hooks', set by 'register_hook')
|
|
PyObject* backward_hooks = nullptr;
|
|
};
|
|
|
|
TORCH_PYTHON_API extern PyObject *THPVariableClass;
|
|
TORCH_PYTHON_API extern PyObject *ParameterClass;
|
|
|
|
TORCH_PYTHON_API PyObject * THPVariable_Wrap(at::TensorBase var);
|
|
|
|
inline bool THPVariable_Check(PyObject *obj)
|
|
{
|
|
if (!THPVariableClass)
|
|
return false;
|
|
|
|
const auto result = PyObject_IsInstance(obj, THPVariableClass);
|
|
AT_ASSERT(result != -1);
|
|
return result;
|
|
}
|
|
|
|
inline const at::Tensor& THPVariable_Unpack(THPVariable* var) {
|
|
return *var->cdata;
|
|
}
|
|
|
|
inline const at::Tensor& THPVariable_Unpack(PyObject* obj) {
|
|
return THPVariable_Unpack(reinterpret_cast<THPVariable*>(obj));
|
|
}
|
|
|
|
TORCH_PYTHON_API c10::impl::PyInterpreter* getPyInterpreter();
|