mirror of https://github.com/llvm/torch-mlir
82 lines
2.6 KiB
MLIR
82 lines
2.6 KiB
MLIR
// RUN: torch-mlir-opt -torch-inline-global-slots -split-input-file %s | FileCheck %s
|
|
|
|
// Transform aspect of the pass.
|
|
|
|
// Test case: Most basic case that can be inlined.
|
|
|
|
// CHECK-NOT: @slot0
|
|
torch.global_slot "private" @slot0 : !torch.int
|
|
|
|
// CHECK-LABEL: torch.global_slot.module_initializer {
|
|
// CHECK: torch.initialize.global_slots [
|
|
// CHECK-NEXT ]
|
|
torch.global_slot.module_initializer {
|
|
%0 = torch.constant.int 1
|
|
torch.initialize.global_slots [
|
|
@slot0(%0 : !torch.int)
|
|
]
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @forward() {
|
|
// CHECK: %[[C1:.*]] = torch.constant.int 1
|
|
// CHECK: return
|
|
func.func @forward() {
|
|
%0 = torch.global_slot.get @slot0 : !torch.int
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// Test case: Shared objects in object graph shared between two initial values.
|
|
|
|
torch.global_slot "private" @tensor : !torch.tensor
|
|
torch.global_slot "private" @list_of_tensor : !torch.list<tensor>
|
|
|
|
// CHECK-LABEL: torch.global_slot.module_initializer {
|
|
// CHECK: torch.initialize.global_slots [
|
|
// CHECK-NEXT ]
|
|
torch.global_slot.module_initializer {
|
|
%0 = torch.tensor.literal(dense<0.0> : tensor<f32>) : !torch.tensor
|
|
%1 = torch.prim.ListConstruct %0 : (!torch.tensor) -> !torch.list<tensor>
|
|
torch.initialize.global_slots [
|
|
@tensor(%0 : !torch.tensor)
|
|
@list_of_tensor(%1 : !torch.list<tensor>)
|
|
]
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @forward() {
|
|
// CHECK: %[[T0:.*]] = torch.tensor.literal(dense<0.000000e+00> : tensor<f32>) : !torch.tensor
|
|
// CHECK: %[[T1:.*]] = torch.tensor.literal(dense<0.000000e+00> : tensor<f32>) : !torch.tensor
|
|
// CHECK: %[[LIST:.*]] = torch.prim.ListConstruct %[[T1]] : (!torch.tensor) -> !torch.list<tensor>
|
|
// CHECK: return
|
|
func.func @forward() {
|
|
%0 = torch.global_slot.get @tensor : !torch.tensor
|
|
%1 = torch.global_slot.get @list_of_tensor : !torch.tensor
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// Test case: Adjusting static info.
|
|
|
|
// CHECK-NOT: @tensor
|
|
torch.global_slot "private" @tensor : !torch.tensor
|
|
|
|
// CHECK-LABEL: torch.global_slot.module_initializer {
|
|
// CHECK: torch.initialize.global_slots [
|
|
// CHECK-NEXT ]
|
|
torch.global_slot.module_initializer {
|
|
%0 = torch.tensor.literal(dense<0.0> : tensor<f32>) : !torch.tensor<[],f32>
|
|
torch.initialize.global_slots [
|
|
@tensor(%0 : !torch.tensor<[],f32>)
|
|
]
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @forward() {
|
|
// CHECK: %[[T:.*]] = torch.tensor.literal(dense<0.000000e+00> : tensor<f32>) : !torch.tensor<[],f32>
|
|
// CHECK: %[[CASTED:.*]] = torch.tensor_static_info_cast %[[T]] : !torch.tensor<[],f32> to !torch.tensor
|
|
func.func @forward() {
|
|
%0 = torch.global_slot.get @tensor : !torch.tensor
|
|
return
|
|
}
|