Go from jit::ClassDef to python code?

DBraun · March 29, 2022, 6:07am

I’m looking at this example

pytorch/pytorch/blob/master/test/cpp/jit/test_class_parser.cpp

#include <gtest/gtest.h>

#include <torch/csrc/jit/frontend/parser.h>
#include <torch/csrc/jit/frontend/resolver.h>

namespace torch {
namespace jit {
constexpr c10::string_view testSource = R"JIT(
  class FooTest:
    def __init__(self, x):
      self.x = x

    def get_x(self):
      return self.x

    an_attribute : Tensor
)JIT";

TEST(ClassParserTest, Basic) {
  Parser p(std::make_shared<Source>(testSource));

This file has been truncated. show original

Is it possible to get Python code from the classDef, to loosely recover the testSource? I read pytorch/serialization.md at master · pytorch/pytorch · GitHub but don’t know where to go for C++ code. This is something close pytorch/script_init.cpp at 51e50a2ddb9844e0abaf6028fe5e4246448dffff · pytorch/pytorch · GitHub

Should I copy this code into a function to achieve my goal? Right now it’s just in a python binding that I can’t use.

Suppose I do that, then I will have an instance of a ClassDef.

Then I could try PythonPrint to print the Python code: https://github.com/pytorch/pytorch/blob/91ef3c82615d6ede05d5b86f1bd5571ea95e4ef1/torch/csrc/jit/python/script_init.cpp#L1235-L1243 I noticed that PythonPrint is missing printClass whereas PythonPrintImpl does have printClass. So I would have to use PythonPrintImpl or duplicate its code, both seem wrong to do.