llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/frontends/pytorch/test/ivalue_import/strings.py

33 lines
928 B
Python
Raw Normal View History

[torch] Add import support for IValue string Type(s) (#179) * [torch] Add import support for IValue string Type(s) * [test] Add test for Strings import
2021-03-05 05:08:50 +08:00
# -*- Python -*-
# This file is licensed under a pytorch-style license
# See frontends/pytorch/LICENSE for license information.
import typing
import torch
import torch_mlir
# RUN: %PYTHON %s | npcomp-opt | FileCheck %s
mb = torch_mlir.ModuleBuilder()
class TestModule(torch.nn.Module):
def __init__(self):
super().__init__()
self.s = "foo"
# CHECK: torch.class_type @[[CLASSTYPE:.*]] {
# TODO: Don't lose element type.
Add !torch.str type. - Remove dependence on `!basicpy.BytesType`. - Add `torch.constant.str "s"` analogous to `torch.constant.none`.
2021-06-15 23:29:06 +08:00
# CHECK: torch.attr "s" : !torch.str
[torch] Add import support for IValue string Type(s) (#179) * [torch] Add import support for IValue string Type(s) * [test] Add test for Strings import
2021-03-05 05:08:50 +08:00
# CHECK: }
Add !torch.str type. - Remove dependence on `!basicpy.BytesType`. - Add `torch.constant.str "s"` analogous to `torch.constant.none`.
2021-06-15 23:29:06 +08:00
# CHECK: %[[STR:.*]] = torch.constant.str "foo"
[torch] Add import support for IValue string Type(s) (#179) * [torch] Add import support for IValue string Type(s) * [test] Add test for Strings import
2021-03-05 05:08:50 +08:00
# CHECK: torch.nn_module {
Add !torch.str type. - Remove dependence on `!basicpy.BytesType`. - Add `torch.constant.str "s"` analogous to `torch.constant.none`.
2021-06-15 23:29:06 +08:00
# CHECK: torch.slot "s", %[[STR]] : !torch.str
[torch] Add import support for IValue string Type(s) (#179) * [torch] Add import support for IValue string Type(s) * [test] Add test for Strings import
2021-03-05 05:08:50 +08:00
# CHECK: } : !torch.nn.Module<"[[CLASSTYPE]]">
test_module = TestModule()
recursivescriptmodule = torch.jit.script(test_module)
# TODO: Automatically handle unpacking Python class RecursiveScriptModule into the underlying ScriptModule.
mb.import_module(recursivescriptmodule._c)
mb.module.operation.print()