torch-mlir/test/Dialect/Torch/GlobalizeObjectGraph/module-uses-error.mlir

14 lines
531 B
MLIR

// RUN: torch-mlir-opt -torch-globalize-object-graph -split-input-file -verify-diagnostics %s
torch.class_type @parent {
torch.method "module_type_return", @module_type_return
}
func.func private @module_type_return(%arg0: !torch.nn.Module<"parent">) {
// expected-error @+1 {{unsupported use of a torch.nn.Module. Expected only method calls or attribute get/set}}
torch.prim.ListConstruct %arg0 : (!torch.nn.Module<"parent">) -> !torch.list<nn.Module<"parent">>
return
}
torch.nn_module {} : !torch.nn.Module<"parent">