mirror of https://github.com/llvm/torch-mlir
21 lines
1.2 KiB
MLIR
21 lines
1.2 KiB
MLIR
|
|
// RUN: npcomp-opt <%s -convert-torch-to-iree -split-input-file -verify-diagnostics | FileCheck %s
|
|
|
|
// CHECK-LABEL: func @forward(
|
|
// CHECK-SAME: %[[ARG_TORCH:.*]]: !torch.float) -> !torch.list<!torch.float> {
|
|
// CHECK: %[[ARG:.*]] = torch_c.to_f64 %[[ARG_TORCH]]
|
|
// CHECK: %[[ALSO_ARG:.*]] = torch_c.to_f64 %[[ARG_TORCH]]
|
|
// CHECK: %[[C2:.*]] = constant 2 : index
|
|
// CHECK: %[[LIST:.*]] = iree.list.create %[[C2]] : !iree.list<f64>
|
|
// CHECK: iree.list.resize %[[LIST]], %[[C2]] : !iree.list<f64>
|
|
// CHECK: %[[C0:.*]] = constant 0 : index
|
|
// CHECK: iree.list.set %[[LIST]][%[[C0]]], %[[ARG]] : !iree.list<f64>, f64
|
|
// CHECK: %[[C1:.*]] = constant 1 : index
|
|
// CHECK: iree.list.set %[[LIST]][%[[C1]]], %[[ALSO_ARG]] : !iree.list<f64>, f64
|
|
// CHECK: %[[LIST_TORCH:.*]] = torch_c.from_iree_list %[[LIST]] : !iree.list<f64> -> !torch.list<!torch.float>
|
|
// CHECK: return %[[LIST_TORCH]] : !torch.list<!torch.float>
|
|
builtin.func @forward(%arg0: !torch.float) -> !torch.list<!torch.float> {
|
|
%0 = torch.prim.ListConstruct %arg0, %arg0 : (!torch.float, !torch.float) -> !torch.list<!torch.float>
|
|
return %0 : !torch.list<!torch.float>
|
|
}
|