mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
dbe850af5b
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/33851 Rationale and context described in #33828. Script to reproduce the move: https://gist.github.com/suo/16cbefaaeb67ca5a7c6caffd49b7f6e9 ghstack-source-id: 99079645 Test Plan: Make sure CI passes Reviewed By: jamesr66a Differential Revision: D20133869 fbshipit-source-id: 390e9241a9c85366d9005c492ac31f10aa96488e
45 lines
1.1 KiB
C++
45 lines
1.1 KiB
C++
#include <torch/csrc/jit/api/object.h>
|
|
|
|
#include <ATen/core/jit_type.h>
|
|
#include <torch/csrc/jit/api/compilation_unit.h>
|
|
#include <torch/csrc/jit/frontend/resolver.h>
|
|
#include <torch/csrc/jit/frontend/sugared_value.h>
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
namespace script {
|
|
|
|
Object::Object(
|
|
std::shared_ptr<CompilationUnit> cu,
|
|
const c10::ClassTypePtr& type)
|
|
: Object(c10::ivalue::Object::create(
|
|
c10::StrongTypePtr(std::move(cu), type),
|
|
type->numAttributes())) {}
|
|
|
|
ObjectPtr Object::_ivalue() const {
|
|
TORCH_INTERNAL_ASSERT(_ivalue_);
|
|
return _ivalue_;
|
|
}
|
|
|
|
c10::optional<Method> Object::find_method(const std::string& basename) const {
|
|
for (Function* fn : type()->methods()) {
|
|
if (fn->name() == basename) {
|
|
return Method(_ivalue(), fn);
|
|
}
|
|
}
|
|
return c10::nullopt;
|
|
}
|
|
|
|
void Object::define(const std::string& src, const ResolverPtr& resolver) {
|
|
const auto self = SimpleSelf(type());
|
|
_ivalue()->compilation_unit()->define(
|
|
*type()->name(),
|
|
src,
|
|
resolver ? resolver : script::nativeResolver(),
|
|
&self);
|
|
}
|
|
|
|
} // namespace script
|
|
} // namespace jit
|
|
} // namespace torch
|