mirror of https://github.com/llvm/torch-mlir
33 lines
1.0 KiB
MLIR
33 lines
1.0 KiB
MLIR
// RUN: torch-mlir-opt -torch-globalize-object-graph -split-input-file %s | FileCheck %s
|
|
|
|
// Check that linkage names consist of the dotted path from the root.
|
|
|
|
// CHECK-LABEL: torch.global_slot.module_initializer {
|
|
// CHECK: %[[FLOAT:.*]] = torch.constant.float 4.200000e+01
|
|
// CHECK: torch.initialize.global_slots [
|
|
// CHECK: @m.float(%[[FLOAT]] : !torch.float)
|
|
// CHECK: ]
|
|
// CHECK: }
|
|
// CHECK-LABEL: torch.global_slot @m.float : !torch.float
|
|
|
|
|
|
torch.class_type @child {
|
|
torch.attr "float" : !torch.float
|
|
}
|
|
torch.class_type @parent {
|
|
torch.attr "m" : !torch.nn.Module<"child">
|
|
}
|
|
|
|
%c42 = torch.constant.float 42.0
|
|
%child = torch.nn_module {
|
|
torch.slot "float", %c42 : !torch.float
|
|
} : !torch.nn.Module<"child">
|
|
%parent = torch.nn_module {
|
|
torch.slot "m", %child : !torch.nn.Module<"child">
|
|
} : !torch.nn.Module<"parent">
|
|
|
|
func.func private @ensure_all_slots_are_used(%arg0: !torch.nn.Module<"child">) {
|
|
%0 = torch.prim.GetAttr %arg0["float"] : !torch.nn.Module<"child"> -> !torch.float
|
|
return
|
|
}
|