mirror of https://github.com/llvm/torch-mlir
Add unit tests
giacs-epic
parent
fd1563a0ff
commit
bac4d775cf
|
|
@ -174,16 +174,412 @@ func.func @torch.aten.fmod_float(%arg0: !torch.vtensor<[?],f16>, %arg1: !torch.v
|
|||
|
||||
// -----
|
||||
|
||||
// CHECK-LABEL: func.func @torch.aten.stft.center(
|
||||
// CHECK-SAME: %[[ARG0:.+]]: !torch.vtensor<[5,35],f32>, %[[ARG1:.+]]: !torch.vtensor<[5],f32>) -> !torch.vtensor<[5,5,16],complex<f32>> {
|
||||
// CHECK: %[[SIGLEN:.+]] = torch.aten.size.int(%[[ARG0:.+]], -1)
|
||||
func.func @torch.aten.stft.center(%arg0: !torch.vtensor<[5,35],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[5,5,16],complex<f32>> {
|
||||
|
||||
// CHECK-LABEL: func.func @torch.aten.stft.center_1D(
|
||||
// CHECK-SAME: %arg0: !torch.vtensor<[40],f32>, %arg1: !torch.vtensor<[4],f32>) -> !torch.vtensor<[3,37],complex<f32>> {
|
||||
// CHECK-DAG: %[[TRUE:.*]] = torch.constant.bool true
|
||||
// CHECK-DAG: %[[INT9:.*]] = torch.constant.int 9
|
||||
// CHECK-DAG: %[[INT4:.*]] = torch.constant.int 4
|
||||
// CHECK-DAG: %[[INT0:.*]] = torch.constant.int 0
|
||||
// CHECK-DAG: %[[INTM1:.*]] = torch.constant.int -1
|
||||
// CHECK-DAG: %[[INT1:.*]] = torch.constant.int 1
|
||||
// CHECK-DAG: %[[NONE:.*]] = torch.constant.none
|
||||
// CHECK-DAG: %float0.000000e00 = torch.constant.float 0.000000e+00
|
||||
// CHECK-DAG: %[[INT40:.*]] = torch.constant.int 40
|
||||
// CHECK-DAG: %[[INT37:.*]] = torch.constant.int 37
|
||||
// CHECK-DAG: %[[INT3:.*]] = torch.constant.int 3
|
||||
// CHECK: %[[VAR0:.*]] = torch.prim.ListConstruct %[[INT3]], %[[INT37]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR1:.*]] = torch.aten.empty.memory_format %[[VAR0]], %[[INT9]], %[[NONE]], %[[NONE]], %[[NONE]], %[[NONE]] : !torch.list<int>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[3,37],complex<f32>>
|
||||
// CHECK: %[[VAR2:.*]] = torch.prim.Loop %[[INT37]], %[[TRUE]], init(%[[VAR1]]) {
|
||||
// CHECK: ^bb0(%arg2: !torch.int, %arg3: !torch.vtensor<[3,37],complex<f32>>):
|
||||
// CHECK: %[[VAR3:.*]] = torch.aten.add.int %arg2, %[[INT4]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR4:.*]] = torch.prim.min.int %[[VAR3]], %[[INT40]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR5:.*]] = torch.aten.sub.int %[[VAR4]], %arg2 : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR6:.*]] = torch.aten.sub.int %[[INT4]], %[[VAR5]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR7:.*]] = torch.aten.add.int %arg2, %[[VAR5]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR8:.*]] = torch.aten.slice.Tensor %arg0, %[[INTM1]], %arg2, %[[VAR7]], %[[INT1]] : !torch.vtensor<[40],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[?],f32>
|
||||
// CHECK: %[[VAR9:.*]] = torch.prim.ListConstruct %[[INT0]], %[[VAR6]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR10:.*]] = torch.aten.constant_pad_nd %[[VAR8]], %[[VAR9]], %float0.000000e00 : !torch.vtensor<[?],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[?],f32>
|
||||
// CHECK: %[[VAR11:.*]] = torch.tensor_static_info_cast %[[VAR10]] : !torch.vtensor<[?],f32> to !torch.vtensor<[4],f32>
|
||||
// CHECK: %[[VAR12:.*]] = torch.aten.mul.Tensor %[[VAR11]], %arg1 : !torch.vtensor<[4],f32>, !torch.vtensor<[4],f32> -> !torch.vtensor<[4],f32>
|
||||
// CHECK: %[[VAR13:.*]] = torch.aten.fft_fft %[[VAR12]], %[[NONE]], %[[INTM1]], %[[NONE]] : !torch.vtensor<[4],f32>, !torch.none, !torch.int, !torch.none -> !torch.vtensor<[3],complex<f32>>
|
||||
// CHECK: %[[VAR14:.*]] = torch.aten.unsqueeze %[[VAR13]], %[[INTM1]] : !torch.vtensor<[3],complex<f32>>, !torch.int -> !torch.vtensor<[3,1],complex<f32>>
|
||||
// CHECK: %[[VAR15:.*]] = torch.aten.slice_scatter %arg3, %[[VAR14]], %[[INTM1]], %arg2, %[[NONE]], %[[INT1]] : !torch.vtensor<[3,37],complex<f32>>, !torch.vtensor<[3,1],complex<f32>>, !torch.int, !torch.int, !torch.none, !torch.int -> !torch.vtensor<[3,37],complex<f32>>
|
||||
// CHECK: torch.prim.Loop.condition %[[TRUE]], iter(%[[VAR15]] : !torch.vtensor<[3,37],complex<f32>>)
|
||||
// CHECK: } : (!torch.int, !torch.bool, !torch.vtensor<[3,37],complex<f32>>) -> !torch.vtensor<[3,37],complex<f32>>
|
||||
// CHECK: return %[[VAR2]] : !torch.vtensor<[3,37],complex<f32>>
|
||||
func.func @torch.aten.stft.center_1D(%arg0: !torch.vtensor<[40],f32>, %arg1: !torch.vtensor<[4],f32>) -> !torch.vtensor<[3,37],complex<f32>> {
|
||||
%padmode = torch.constant.str "reflect"
|
||||
%nfft = torch.constant.int 5
|
||||
%hoplen = torch.constant.int 2
|
||||
%nfft = torch.constant.int 4
|
||||
%hoplen = torch.constant.int 1
|
||||
%winlen = torch.constant.int 4
|
||||
%cstfalse = torch.constant.bool false
|
||||
%csttrue = torch.constant.bool true
|
||||
%0 = torch.aten.stft.center %arg0, %nfft, %hoplen, %winlen, %arg1, %cstfalse, %padmode, %cstfalse, %cstfalse, %csttrue : !torch.vtensor<[5,35],f32>, !torch.int, !torch.int, !torch.int, !torch.vtensor<[5],f32>, !torch.bool, !torch.str, !torch.bool, !torch.bool, !torch.bool -> !torch.vtensor<[5,5,16],complex<f32>>
|
||||
return %0 : !torch.vtensor<[5,5,16],complex<f32>>
|
||||
%0 = torch.aten.stft.center %arg0, %nfft, %hoplen, %winlen, %arg1, %cstfalse, %padmode, %cstfalse, %cstfalse, %csttrue : !torch.vtensor<[40],f32>, !torch.int, !torch.int, !torch.int, !torch.vtensor<[4],f32>, !torch.bool, !torch.str, !torch.bool, !torch.bool, !torch.bool -> !torch.vtensor<[3,37],complex<f32>>
|
||||
return %0 : !torch.vtensor<[3,37],complex<f32>>
|
||||
}
|
||||
|
||||
|
||||
// -----
|
||||
|
||||
|
||||
// CHECK-LABEL: func.func @torch.aten.stft.center_1D_unk_sig_len(
|
||||
// CHECK-SAME: %arg0: !torch.vtensor<[?],f32>, %arg1: !torch.vtensor<[10],f32>) -> !torch.vtensor<[6,?],complex<f32>> {
|
||||
// CHECK-DAG: %[[TRUE:.*]] = torch.constant.bool true
|
||||
// CHECK-DAG: %[[INT9:.*]] = torch.constant.int 9
|
||||
// CHECK-DAG: %[[INT6:.*]] = torch.constant.int 6
|
||||
// CHECK-DAG: %[[INT10:.*]] = torch.constant.int 10
|
||||
// CHECK-DAG: %[[INT1:.*]] = torch.constant.int 1
|
||||
// CHECK-DAG: %[[INT0:.*]] = torch.constant.int 0
|
||||
// CHECK-DAG: %[[INTM1:.*]] = torch.constant.int -1
|
||||
// CHECK-DAG: %[[NONE:.*]] = torch.constant.none
|
||||
// CHECK-DAG: %float0.000000e00 = torch.constant.float 0.000000e+00
|
||||
// CHECK: %[[VAR0:.*]] = torch.aten.size.int %arg0, %[[INTM1]] : !torch.vtensor<[?],f32>, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR1:.*]] = torch.aten.sub.int %[[VAR0]], %[[INT10]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR2:.*]] = torch.aten.floordiv.int %[[VAR1]], %[[INT1]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR3:.*]] = torch.aten.add.int %[[INT1]], %[[VAR2]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR4:.*]] = torch.prim.ListConstruct %[[INT6]], %[[VAR3]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR5:.*]] = torch.aten.empty.memory_format %[[VAR4]], %[[INT9]], %[[NONE]], %[[NONE]], %[[NONE]], %[[NONE]] : !torch.list<int>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[6,?],complex<f32>>
|
||||
// CHECK: %[[VAR6:.*]] = torch.prim.Loop %[[VAR3]], %[[TRUE]], init(%[[VAR5]]) {
|
||||
// CHECK: ^bb0(%arg2: !torch.int, %arg3: !torch.vtensor<[6,?],complex<f32>>):
|
||||
// CHECK: %[[VAR7:.*]] = torch.aten.add.int %arg2, %[[INT10]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR8:.*]] = torch.prim.min.int %[[VAR7]], %[[VAR0]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR9:.*]] = torch.aten.sub.int %[[VAR8]], %arg2 : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR10:.*]] = torch.aten.sub.int %[[INT10]], %[[VAR9]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR11:.*]] = torch.aten.add.int %arg2, %[[VAR9]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR12:.*]] = torch.aten.slice.Tensor %arg0, %[[INTM1]], %arg2, %[[VAR11]], %[[INT1]] : !torch.vtensor<[?],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[?],f32>
|
||||
// CHECK: %[[VAR13:.*]] = torch.prim.ListConstruct %[[INT0]], %[[VAR10]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR14:.*]] = torch.aten.constant_pad_nd %[[VAR12]], %[[VAR13]], %float0.000000e00 : !torch.vtensor<[?],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[?],f32>
|
||||
// CHECK: %[[VAR15:.*]] = torch.tensor_static_info_cast %[[VAR14]] : !torch.vtensor<[?],f32> to !torch.vtensor<[10],f32>
|
||||
// CHECK: %[[VAR16:.*]] = torch.aten.mul.Tensor %[[VAR15]], %arg1 : !torch.vtensor<[10],f32>, !torch.vtensor<[10],f32> -> !torch.vtensor<[10],f32>
|
||||
// CHECK: %[[VAR17:.*]] = torch.aten.fft_fft %[[VAR16]], %[[NONE]], %[[INTM1]], %[[NONE]] : !torch.vtensor<[10],f32>, !torch.none, !torch.int, !torch.none -> !torch.vtensor<[6],complex<f32>>
|
||||
// CHECK: %[[VAR18:.*]] = torch.aten.unsqueeze %[[VAR17]], %[[INTM1]] : !torch.vtensor<[6],complex<f32>>, !torch.int -> !torch.vtensor<[6,1],complex<f32>>
|
||||
// CHECK: %[[VAR19:.*]] = torch.aten.slice_scatter %arg3, %[[VAR18]], %[[INTM1]], %arg2, %[[NONE]], %[[INT1]] : !torch.vtensor<[6,?],complex<f32>>, !torch.vtensor<[6,1],complex<f32>>, !torch.int, !torch.int, !torch.none, !torch.int -> !torch.vtensor<[6,?],complex<f32>>
|
||||
// CHECK: torch.prim.Loop.condition %[[TRUE]], iter(%[[VAR19]] : !torch.vtensor<[6,?],complex<f32>>)
|
||||
// CHECK: } : (!torch.int, !torch.bool, !torch.vtensor<[6,?],complex<f32>>) -> !torch.vtensor<[6,?],complex<f32>>
|
||||
// CHECK: return %[[VAR6]] : !torch.vtensor<[6,?],complex<f32>>
|
||||
func.func @torch.aten.stft.center_1D_unk_sig_len(%arg0: !torch.vtensor<[?],f32>, %arg1: !torch.vtensor<[10],f32>) -> !torch.vtensor<[6,?],complex<f32>> {
|
||||
%padmode = torch.constant.str "reflect"
|
||||
%nfft = torch.constant.int 10
|
||||
%hoplen = torch.constant.int 1
|
||||
%winlen = torch.constant.int 10
|
||||
%cstfalse = torch.constant.bool false
|
||||
%csttrue = torch.constant.bool true
|
||||
%0 = torch.aten.stft.center %arg0, %nfft, %hoplen, %winlen, %arg1, %cstfalse, %padmode, %cstfalse, %cstfalse, %csttrue : !torch.vtensor<[?],f32>, !torch.int, !torch.int, !torch.int, !torch.vtensor<[10],f32>, !torch.bool, !torch.str, !torch.bool, !torch.bool, !torch.bool -> !torch.vtensor<[6,?],complex<f32>>
|
||||
return %0 : !torch.vtensor<[6,?],complex<f32>>
|
||||
}
|
||||
|
||||
|
||||
// -----
|
||||
|
||||
|
||||
// CHECK-LABEL: func.func @torch.aten.stft.center_2D(
|
||||
// CHECK-SAME: %arg0: !torch.vtensor<[4,46],f32>, %arg1: !torch.vtensor<[7],f32>) -> !torch.vtensor<[4,4,40],complex<f32>> {
|
||||
// CHECK-DAG: %[[TRUE:.*]] = torch.constant.bool true
|
||||
// CHECK-DAG: %[[INT9:.*]] = torch.constant.int 9
|
||||
// CHECK-DAG: %[[INT4:.*]] = torch.constant.int 4
|
||||
// CHECK-DAG: %[[INT7:.*]] = torch.constant.int 7
|
||||
// CHECK-DAG: %[[INT0:.*]] = torch.constant.int 0
|
||||
// CHECK-DAG: %[[INTM1:.*]] = torch.constant.int -1
|
||||
// CHECK-DAG: %[[INT1:.*]] = torch.constant.int 1
|
||||
// CHECK-DAG: %[[NONE:.*]] = torch.constant.none
|
||||
// CHECK-DAG: %float0.000000e00 = torch.constant.float 0.000000e+00
|
||||
// CHECK-DAG: %[[INT46:.*]] = torch.constant.int 46
|
||||
// CHECK-DAG: %[[INT40:.*]] = torch.constant.int 40
|
||||
// CHECK: %[[VAR0:.*]] = torch.prim.ListConstruct %[[INT1]], %[[INT7]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR1:.*]] = torch.aten.view %arg1, %[[VAR0]] : !torch.vtensor<[7],f32>, !torch.list<int> -> !torch.vtensor<[1,7],f32>
|
||||
// CHECK: %[[VAR2:.*]] = torch.prim.ListConstruct %[[INT4]], %[[INT4]], %[[INT40]] : (!torch.int, !torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR3:.*]] = torch.aten.empty.memory_format %[[VAR2]], %[[INT9]], %[[NONE]], %[[NONE]], %[[NONE]], %[[NONE]] : !torch.list<int>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[4,4,40],complex<f32>>
|
||||
// CHECK: %[[VAR4:.*]] = torch.prim.Loop %[[INT40]], %[[TRUE]], init(%[[VAR3]]) {
|
||||
// CHECK: ^bb0(%arg2: !torch.int, %arg3: !torch.vtensor<[4,4,40],complex<f32>>):
|
||||
// CHECK: %[[VAR5:.*]] = torch.aten.add.int %arg2, %[[INT7]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR6:.*]] = torch.prim.min.int %[[VAR5]], %[[INT46]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR7:.*]] = torch.aten.sub.int %[[VAR6]], %arg2 : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR8:.*]] = torch.aten.sub.int %[[INT7]], %[[VAR7]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR9:.*]] = torch.aten.add.int %arg2, %[[VAR7]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR10:.*]] = torch.aten.slice.Tensor %arg0, %[[INTM1]], %arg2, %[[VAR9]], %[[INT1]] : !torch.vtensor<[4,46],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[4,?],f32>
|
||||
// CHECK: %[[VAR11:.*]] = torch.prim.ListConstruct %[[INT0]], %[[VAR8]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR12:.*]] = torch.aten.constant_pad_nd %[[VAR10]], %[[VAR11]], %float0.000000e00 : !torch.vtensor<[4,?],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[4,?],f32>
|
||||
// CHECK: %[[VAR13:.*]] = torch.tensor_static_info_cast %[[VAR12]] : !torch.vtensor<[4,?],f32> to !torch.vtensor<[4,7],f32>
|
||||
// CHECK: %[[VAR14:.*]] = torch.aten.mul.Tensor %[[VAR13]], %[[VAR1]] : !torch.vtensor<[4,7],f32>, !torch.vtensor<[1,7],f32> -> !torch.vtensor<[4,7],f32>
|
||||
// CHECK: %[[VAR15:.*]] = torch.aten.fft_fft %[[VAR14]], %[[NONE]], %[[INTM1]], %[[NONE]] : !torch.vtensor<[4,7],f32>, !torch.none, !torch.int, !torch.none -> !torch.vtensor<[4,4],complex<f32>>
|
||||
// CHECK: %[[VAR16:.*]] = torch.aten.unsqueeze %[[VAR15]], %[[INTM1]] : !torch.vtensor<[4,4],complex<f32>>, !torch.int -> !torch.vtensor<[4,4,1],complex<f32>>
|
||||
// CHECK: %[[VAR17:.*]] = torch.aten.slice_scatter %arg3, %[[VAR16]], %[[INTM1]], %arg2, %[[NONE]], %[[INT1]] : !torch.vtensor<[4,4,40],complex<f32>>, !torch.vtensor<[4,4,1],complex<f32>>, !torch.int, !torch.int, !torch.none, !torch.int -> !torch.vtensor<[4,4,40],complex<f32>>
|
||||
// CHECK: torch.prim.Loop.condition %[[TRUE]], iter(%[[VAR17]] : !torch.vtensor<[4,4,40],complex<f32>>)
|
||||
// CHECK: } : (!torch.int, !torch.bool, !torch.vtensor<[4,4,40],complex<f32>>) -> !torch.vtensor<[4,4,40],complex<f32>>
|
||||
// CHECK: return %[[VAR4]] : !torch.vtensor<[4,4,40],complex<f32>>
|
||||
func.func @torch.aten.stft.center_2D(%arg0: !torch.vtensor<[4,46],f32>, %arg1: !torch.vtensor<[7],f32>) -> !torch.vtensor<[4,4,40],complex<f32>> {
|
||||
%padmode = torch.constant.str "reflect"
|
||||
%nfft = torch.constant.int 7
|
||||
%hoplen = torch.constant.int 1
|
||||
%winlen = torch.constant.int 7
|
||||
%cstfalse = torch.constant.bool false
|
||||
%csttrue = torch.constant.bool true
|
||||
%0 = torch.aten.stft.center %arg0, %nfft, %hoplen, %winlen, %arg1, %cstfalse, %padmode, %cstfalse, %cstfalse, %csttrue : !torch.vtensor<[4,46],f32>, !torch.int, !torch.int, !torch.int, !torch.vtensor<[7],f32>, !torch.bool, !torch.str, !torch.bool, !torch.bool, !torch.bool -> !torch.vtensor<[4,4,40],complex<f32>>
|
||||
return %0 : !torch.vtensor<[4,4,40],complex<f32>>
|
||||
}
|
||||
|
||||
|
||||
// -----
|
||||
|
||||
|
||||
// CHECK-LABEL: func.func @torch.aten.stft.center_2D_win_unk_size(
|
||||
// CHECK-SAME: %arg0: !torch.vtensor<[3,38],f32>, %arg1: !torch.vtensor<[?],f32>) -> !torch.vtensor<[3,4,32],complex<f32>> {
|
||||
// CHECK-DAG: %[[TRUE:.*]] = torch.constant.bool true
|
||||
// CHECK-DAG: %[[INT9:.*]] = torch.constant.int 9
|
||||
// CHECK-DAG: %[[INT3:.*]] = torch.constant.int 3
|
||||
// CHECK-DAG: %[[INT4:.*]] = torch.constant.int 4
|
||||
// CHECK-DAG: %[[INT32:.*]] = torch.constant.int 32
|
||||
// CHECK-DAG: %[[INT38:.*]] = torch.constant.int 38
|
||||
// CHECK-DAG: %float0.000000e00 = torch.constant.float 0.000000e+00
|
||||
// CHECK-DAG: %[[NONE:.*]] = torch.constant.none
|
||||
// CHECK-DAG: %[[INT1:.*]] = torch.constant.int 1
|
||||
// CHECK-DAG: %[[INTM1:.*]] = torch.constant.int -1
|
||||
// CHECK-DAG: %[[INT6:.*]] = torch.constant.int 6
|
||||
// CHECK-DAG: %[[INT7:.*]] = torch.constant.int 7
|
||||
// CHECK-DAG: %[[INT0:.*]] = torch.constant.int 0
|
||||
// CHECK: %[[VAR0:.*]] = torch.aten.size.int %arg1, %[[INT0]] : !torch.vtensor<[?],f32>, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR1:.*]] = torch.aten.eq.int %[[VAR0]], %[[INT6]] : !torch.int, !torch.int -> !torch.bool
|
||||
// CHECK: torch.runtime.assert %[[VAR1]], "window size should be equal to win_length"
|
||||
// CHECK: %[[VAR2:.*]] = torch.prim.ListConstruct %[[INT0]], %[[INT1]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR3:.*]] = torch.aten.constant_pad_nd %arg1, %[[VAR2]], %float0.000000e00 : !torch.vtensor<[?],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[7],f32>
|
||||
// CHECK: %[[VAR4:.*]] = torch.prim.ListConstruct %[[INT1]], %[[INT7]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR5:.*]] = torch.aten.view %[[VAR3]], %[[VAR4]] : !torch.vtensor<[7],f32>, !torch.list<int> -> !torch.vtensor<[1,7],f32>
|
||||
// CHECK: %[[VAR6:.*]] = torch.prim.ListConstruct %[[INT3]], %[[INT4]], %[[INT32]] : (!torch.int, !torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR7:.*]] = torch.aten.empty.memory_format %[[VAR6]], %[[INT9]], %[[NONE]], %[[NONE]], %[[NONE]], %[[NONE]] : !torch.list<int>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[3,4,32],complex<f32>>
|
||||
// CHECK: %[[VAR8:.*]] = torch.prim.Loop %[[INT32]], %[[TRUE]], init(%[[VAR7]]) {
|
||||
// CHECK: ^bb0(%arg2: !torch.int, %arg3: !torch.vtensor<[3,4,32],complex<f32>>):
|
||||
// CHECK: %[[VAR9:.*]] = torch.aten.add.int %arg2, %[[INT7]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR10:.*]] = torch.prim.min.int %[[VAR9]], %[[INT38]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR11:.*]] = torch.aten.sub.int %[[VAR10]], %arg2 : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR12:.*]] = torch.aten.sub.int %[[INT7]], %[[VAR11]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR13:.*]] = torch.aten.add.int %arg2, %[[VAR11]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR14:.*]] = torch.aten.slice.Tensor %arg0, %[[INTM1]], %arg2, %[[VAR13]], %[[INT1]] : !torch.vtensor<[3,38],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[3,?],f32>
|
||||
// CHECK: %[[VAR15:.*]] = torch.prim.ListConstruct %[[INT0]], %[[VAR12]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR16:.*]] = torch.aten.constant_pad_nd %[[VAR14]], %[[VAR15]], %float0.000000e00 : !torch.vtensor<[3,?],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[3,?],f32>
|
||||
// CHECK: %[[VAR17:.*]] = torch.tensor_static_info_cast %[[VAR16]] : !torch.vtensor<[3,?],f32> to !torch.vtensor<[3,7],f32>
|
||||
// CHECK: %[[VAR18:.*]] = torch.aten.mul.Tensor %[[VAR17]], %[[VAR5]] : !torch.vtensor<[3,7],f32>, !torch.vtensor<[1,7],f32> -> !torch.vtensor<[3,7],f32>
|
||||
// CHECK: %[[VAR19:.*]] = torch.aten.fft_fft %[[VAR18]], %[[NONE]], %[[INTM1]], %[[NONE]] : !torch.vtensor<[3,7],f32>, !torch.none, !torch.int, !torch.none -> !torch.vtensor<[3,4],complex<f32>>
|
||||
// CHECK: %[[VAR20:.*]] = torch.aten.unsqueeze %[[VAR19]], %[[INTM1]] : !torch.vtensor<[3,4],complex<f32>>, !torch.int -> !torch.vtensor<[3,4,1],complex<f32>>
|
||||
// CHECK: %[[VAR21:.*]] = torch.aten.slice_scatter %arg3, %[[VAR20]], %[[INTM1]], %arg2, %[[NONE]], %[[INT1]] : !torch.vtensor<[3,4,32],complex<f32>>, !torch.vtensor<[3,4,1],complex<f32>>, !torch.int, !torch.int, !torch.none, !torch.int -> !torch.vtensor<[3,4,32],complex<f32>>
|
||||
// CHECK: torch.prim.Loop.condition %[[TRUE]], iter(%[[VAR21]] : !torch.vtensor<[3,4,32],complex<f32>>)
|
||||
// CHECK: } : (!torch.int, !torch.bool, !torch.vtensor<[3,4,32],complex<f32>>) -> !torch.vtensor<[3,4,32],complex<f32>>
|
||||
// CHECK: return %[[VAR8]] : !torch.vtensor<[3,4,32],complex<f32>>
|
||||
func.func @torch.aten.stft.center_2D_win_unk_size(%arg0: !torch.vtensor<[3,38],f32>, %arg1: !torch.vtensor<[?],f32>) -> !torch.vtensor<[3,4,32],complex<f32>> {
|
||||
%padmode = torch.constant.str "reflect"
|
||||
%nfft = torch.constant.int 7
|
||||
%hoplen = torch.constant.int 1
|
||||
%winlen = torch.constant.int 6
|
||||
%cstfalse = torch.constant.bool false
|
||||
%csttrue = torch.constant.bool true
|
||||
%0 = torch.aten.stft.center %arg0, %nfft, %hoplen, %winlen, %arg1, %cstfalse, %padmode, %cstfalse, %cstfalse, %csttrue : !torch.vtensor<[3,38],f32>, !torch.int, !torch.int, !torch.int, !torch.vtensor<[?],f32>, !torch.bool, !torch.str, !torch.bool, !torch.bool, !torch.bool -> !torch.vtensor<[3,4,32],complex<f32>>
|
||||
return %0 : !torch.vtensor<[3,4,32],complex<f32>>
|
||||
}
|
||||
|
||||
|
||||
// -----
|
||||
|
||||
|
||||
// CHECK-LABEL: func.func @torch.aten.stft.center_2D_no_window(
|
||||
// CHECK-SAME: %arg0: !torch.vtensor<[2,32],f32>) -> !torch.vtensor<[2,5,25],complex<f32>> {
|
||||
// CHECK-DAG: %[[TRUE:.*]] = torch.constant.bool true
|
||||
// CHECK-DAG: %[[INT9:.*]] = torch.constant.int 9
|
||||
// CHECK-DAG: %[[INT8:.*]] = torch.constant.int 8
|
||||
// CHECK-DAG: %[[INT0:.*]] = torch.constant.int 0
|
||||
// CHECK-DAG: %[[INTM1:.*]] = torch.constant.int -1
|
||||
// CHECK-DAG: %[[INT1:.*]] = torch.constant.int 1
|
||||
// CHECK-DAG: %[[NONE:.*]] = torch.constant.none
|
||||
// CHECK-DAG: %float0.000000e00 = torch.constant.float 0.000000e+00
|
||||
// CHECK-DAG: %[[INT32:.*]] = torch.constant.int 32
|
||||
// CHECK-DAG: %[[INT25:.*]] = torch.constant.int 25
|
||||
// CHECK-DAG: %[[INT5:.*]] = torch.constant.int 5
|
||||
// CHECK-DAG: %[[INT2:.*]] = torch.constant.int 2
|
||||
// CHECK: %[[VAR0:.*]] = torch.prim.ListConstruct %[[INT2]], %[[INT5]], %[[INT25]] : (!torch.int, !torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR1:.*]] = torch.aten.empty.memory_format %[[VAR0]], %[[INT9]], %[[NONE]], %[[NONE]], %[[NONE]], %[[NONE]] : !torch.list<int>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[2,5,25],complex<f32>>
|
||||
// CHECK: %[[VAR2:.*]] = torch.prim.Loop %[[INT25]], %[[TRUE]], init(%[[VAR1]]) {
|
||||
// CHECK: ^bb0(%arg1: !torch.int, %arg2: !torch.vtensor<[2,5,25],complex<f32>>):
|
||||
// CHECK: %[[VAR3:.*]] = torch.aten.add.int %arg1, %[[INT8]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR4:.*]] = torch.prim.min.int %[[VAR3]], %[[INT32]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR5:.*]] = torch.aten.sub.int %[[VAR4]], %arg1 : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR6:.*]] = torch.aten.sub.int %[[INT8]], %[[VAR5]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR7:.*]] = torch.aten.add.int %arg1, %[[VAR5]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR8:.*]] = torch.aten.slice.Tensor %arg0, %[[INTM1]], %arg1, %[[VAR7]], %[[INT1]] : !torch.vtensor<[2,32],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[2,?],f32>
|
||||
// CHECK: %[[VAR9:.*]] = torch.prim.ListConstruct %[[INT0]], %[[VAR6]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR10:.*]] = torch.aten.constant_pad_nd %[[VAR8]], %[[VAR9]], %float0.000000e00 : !torch.vtensor<[2,?],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[2,?],f32>
|
||||
// CHECK: %[[VAR11:.*]] = torch.tensor_static_info_cast %[[VAR10]] : !torch.vtensor<[2,?],f32> to !torch.vtensor<[2,8],f32>
|
||||
// CHECK: %[[VAR12:.*]] = torch.aten.fft_fft %[[VAR11]], %[[NONE]], %[[INTM1]], %[[NONE]] : !torch.vtensor<[2,8],f32>, !torch.none, !torch.int, !torch.none -> !torch.vtensor<[2,5],complex<f32>>
|
||||
// CHECK: %[[VAR13:.*]] = torch.aten.unsqueeze %[[VAR12]], %[[INTM1]] : !torch.vtensor<[2,5],complex<f32>>, !torch.int -> !torch.vtensor<[2,5,1],complex<f32>>
|
||||
// CHECK: %[[VAR14:.*]] = torch.aten.slice_scatter %arg2, %[[VAR13]], %[[INTM1]], %arg1, %[[NONE]], %[[INT1]] : !torch.vtensor<[2,5,25],complex<f32>>, !torch.vtensor<[2,5,1],complex<f32>>, !torch.int, !torch.int, !torch.none, !torch.int -> !torch.vtensor<[2,5,25],complex<f32>>
|
||||
// CHECK: torch.prim.Loop.condition %[[TRUE]], iter(%[[VAR14]] : !torch.vtensor<[2,5,25],complex<f32>>)
|
||||
// CHECK: } : (!torch.int, !torch.bool, !torch.vtensor<[2,5,25],complex<f32>>) -> !torch.vtensor<[2,5,25],complex<f32>>
|
||||
// CHECK: return %[[VAR2]] : !torch.vtensor<[2,5,25],complex<f32>>
|
||||
func.func @torch.aten.stft.center_2D_no_window(%arg0: !torch.vtensor<[2,32],f32>) -> !torch.vtensor<[2,5,25],complex<f32>> {
|
||||
%padmode = torch.constant.str "reflect"
|
||||
%nfft = torch.constant.int 8
|
||||
%hoplen = torch.constant.int 1
|
||||
%cstfalse = torch.constant.bool false
|
||||
%csttrue = torch.constant.bool true
|
||||
%cstnone = torch.constant.none
|
||||
%0 = torch.aten.stft.center %arg0, %nfft, %hoplen, %cstnone, %cstnone, %cstfalse, %padmode, %cstfalse, %cstfalse, %csttrue : !torch.vtensor<[2,32],f32>, !torch.int, !torch.int, !torch.none, !torch.none, !torch.bool, !torch.str, !torch.bool, !torch.bool, !torch.bool -> !torch.vtensor<[2,5,25],complex<f32>>
|
||||
return %0 : !torch.vtensor<[2,5,25],complex<f32>>
|
||||
}
|
||||
|
||||
|
||||
// -----
|
||||
|
||||
|
||||
// CHECK-LABEL: func.func @torch.aten.stft.center_2D_hop_length_2(
|
||||
// CHECK-SAME: %arg0: !torch.vtensor<[2,61],f32>, %arg1: !torch.vtensor<[8],f32>) -> !torch.vtensor<[2,5,27],complex<f32>> {
|
||||
// CHECK-DAG: %[[TRUE:.*]] = torch.constant.bool true
|
||||
// CHECK-DAG: %[[INT9:.*]] = torch.constant.int 9
|
||||
// CHECK-DAG: %[[INT5:.*]] = torch.constant.int 5
|
||||
// CHECK-DAG: %[[INT8:.*]] = torch.constant.int 8
|
||||
// CHECK-DAG: %[[INT2:.*]] = torch.constant.int 2
|
||||
// CHECK-DAG: %[[INT0:.*]] = torch.constant.int 0
|
||||
// CHECK-DAG: %[[INTM1:.*]] = torch.constant.int -1
|
||||
// CHECK-DAG: %[[INT1:.*]] = torch.constant.int 1
|
||||
// CHECK-DAG: %[[NONE:.*]] = torch.constant.none
|
||||
// CHECK-DAG: %float0.000000e00 = torch.constant.float 0.000000e+00
|
||||
// CHECK-DAG: %[[INT61:.*]] = torch.constant.int 61
|
||||
// CHECK-DAG: %[[INT27:.*]] = torch.constant.int 27
|
||||
// CHECK: %[[VAR0:.*]] = torch.prim.ListConstruct %[[INT1]], %[[INT8]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR1:.*]] = torch.aten.view %arg1, %[[VAR0]] : !torch.vtensor<[8],f32>, !torch.list<int> -> !torch.vtensor<[1,8],f32>
|
||||
// CHECK: %[[VAR2:.*]] = torch.prim.ListConstruct %[[INT2]], %[[INT5]], %[[INT27]] : (!torch.int, !torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR3:.*]] = torch.aten.empty.memory_format %[[VAR2]], %[[INT9]], %[[NONE]], %[[NONE]], %[[NONE]], %[[NONE]] : !torch.list<int>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[2,5,27],complex<f32>>
|
||||
// CHECK: %[[VAR4:.*]] = torch.prim.Loop %[[INT27]], %[[TRUE]], init(%[[VAR3]]) {
|
||||
// CHECK: ^bb0(%arg2: !torch.int, %arg3: !torch.vtensor<[2,5,27],complex<f32>>):
|
||||
// CHECK: %[[VAR5:.*]] = torch.aten.mul.int %arg2, %[[INT2]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR6:.*]] = torch.aten.add.int %[[VAR5]], %[[INT8]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR7:.*]] = torch.prim.min.int %[[VAR6]], %[[INT61]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR8:.*]] = torch.aten.sub.int %[[VAR7]], %[[VAR5]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR9:.*]] = torch.aten.sub.int %[[INT8]], %[[VAR8]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR10:.*]] = torch.aten.add.int %[[VAR5]], %[[VAR8]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR11:.*]] = torch.aten.slice.Tensor %arg0, %[[INTM1]], %[[VAR5]], %[[VAR10]], %[[INT1]] : !torch.vtensor<[2,61],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[2,?],f32>
|
||||
// CHECK: %[[VAR12:.*]] = torch.prim.ListConstruct %[[INT0]], %[[VAR9]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR13:.*]] = torch.aten.constant_pad_nd %[[VAR11]], %[[VAR12]], %float0.000000e00 : !torch.vtensor<[2,?],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[2,?],f32>
|
||||
// CHECK: %[[VAR14:.*]] = torch.tensor_static_info_cast %[[VAR13]] : !torch.vtensor<[2,?],f32> to !torch.vtensor<[2,8],f32>
|
||||
// CHECK: %[[VAR15:.*]] = torch.aten.mul.Tensor %[[VAR14]], %[[VAR1]] : !torch.vtensor<[2,8],f32>, !torch.vtensor<[1,8],f32> -> !torch.vtensor<[2,8],f32>
|
||||
// CHECK: %[[VAR16:.*]] = torch.aten.fft_fft %[[VAR15]], %[[NONE]], %[[INTM1]], %[[NONE]] : !torch.vtensor<[2,8],f32>, !torch.none, !torch.int, !torch.none -> !torch.vtensor<[2,5],complex<f32>>
|
||||
// CHECK: %[[VAR17:.*]] = torch.aten.unsqueeze %[[VAR16]], %[[INTM1]] : !torch.vtensor<[2,5],complex<f32>>, !torch.int -> !torch.vtensor<[2,5,1],complex<f32>>
|
||||
// CHECK: %[[VAR18:.*]] = torch.aten.slice_scatter %arg3, %[[VAR17]], %[[INTM1]], %arg2, %[[NONE]], %[[INT1]] : !torch.vtensor<[2,5,27],complex<f32>>, !torch.vtensor<[2,5,1],complex<f32>>, !torch.int, !torch.int, !torch.none, !torch.int -> !torch.vtensor<[2,5,27],complex<f32>>
|
||||
// CHECK: torch.prim.Loop.condition %[[TRUE]], iter(%[[VAR18]] : !torch.vtensor<[2,5,27],complex<f32>>)
|
||||
// CHECK: } : (!torch.int, !torch.bool, !torch.vtensor<[2,5,27],complex<f32>>) -> !torch.vtensor<[2,5,27],complex<f32>>
|
||||
// CHECK: return %[[VAR4]] : !torch.vtensor<[2,5,27],complex<f32>>
|
||||
func.func @torch.aten.stft.center_2D_hop_length_2(%arg0: !torch.vtensor<[2,61],f32>, %arg1: !torch.vtensor<[8],f32>) -> !torch.vtensor<[2,5,27],complex<f32>> {
|
||||
%padmode = torch.constant.str "reflect"
|
||||
%nfft = torch.constant.int 8
|
||||
%hoplen = torch.constant.int 2
|
||||
%winlen = torch.constant.int 8
|
||||
%cstfalse = torch.constant.bool false
|
||||
%csttrue = torch.constant.bool true
|
||||
%0 = torch.aten.stft.center %arg0, %nfft, %hoplen, %winlen, %arg1, %cstfalse, %padmode, %cstfalse, %cstfalse, %csttrue : !torch.vtensor<[2,61],f32>, !torch.int, !torch.int, !torch.int, !torch.vtensor<[8],f32>, !torch.bool, !torch.str, !torch.bool, !torch.bool, !torch.bool -> !torch.vtensor<[2,5,27],complex<f32>>
|
||||
return %0 : !torch.vtensor<[2,5,27],complex<f32>>
|
||||
}
|
||||
|
||||
|
||||
// -----
|
||||
|
||||
|
||||
// CHECK-LABEL: func.func @torch.aten.stft.center_2D_window_pad_left(
|
||||
// CHECK-SAME: %arg0: !torch.vtensor<[2,68],f32>, %arg1: !torch.vtensor<[6],f32>) -> !torch.vtensor<[2,4,31],complex<f32>> {
|
||||
// CHECK-DAG: %[[TRUE:.*]] = torch.constant.bool true
|
||||
// CHECK-DAG: %[[INT9:.*]] = torch.constant.int 9
|
||||
// CHECK-DAG: %[[INT4:.*]] = torch.constant.int 4
|
||||
// CHECK-DAG: %[[INT7:.*]] = torch.constant.int 7
|
||||
// CHECK-DAG: %[[INT2:.*]] = torch.constant.int 2
|
||||
// CHECK-DAG: %[[INT0:.*]] = torch.constant.int 0
|
||||
// CHECK-DAG: %[[INTM1:.*]] = torch.constant.int -1
|
||||
// CHECK-DAG: %[[INT1:.*]] = torch.constant.int 1
|
||||
// CHECK-DAG: %[[NONE:.*]] = torch.constant.none
|
||||
// CHECK-DAG: %float0.000000e00 = torch.constant.float 0.000000e+00
|
||||
// CHECK-DAG: %[[INT68:.*]] = torch.constant.int 68
|
||||
// CHECK-DAG: %[[INT31:.*]] = torch.constant.int 31
|
||||
// CHECK: %[[VAR0:.*]] = torch.prim.ListConstruct %[[INT0]], %[[INT1]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR1:.*]] = torch.aten.constant_pad_nd %arg1, %[[VAR0]], %float0.000000e00 : !torch.vtensor<[6],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[7],f32>
|
||||
// CHECK: %[[VAR2:.*]] = torch.prim.ListConstruct %[[INT1]], %[[INT7]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR3:.*]] = torch.aten.view %[[VAR1]], %[[VAR2]] : !torch.vtensor<[7],f32>, !torch.list<int> -> !torch.vtensor<[1,7],f32>
|
||||
// CHECK: %[[VAR4:.*]] = torch.prim.ListConstruct %[[INT2]], %[[INT4]], %[[INT31]] : (!torch.int, !torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR5:.*]] = torch.aten.empty.memory_format %[[VAR4]], %[[INT9]], %[[NONE]], %[[NONE]], %[[NONE]], %[[NONE]] : !torch.list<int>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[2,4,31],complex<f32>>
|
||||
// CHECK: %[[VAR6:.*]] = torch.prim.Loop %[[INT31]], %[[TRUE]], init(%[[VAR5]]) {
|
||||
// CHECK: ^bb0(%arg2: !torch.int, %arg3: !torch.vtensor<[2,4,31],complex<f32>>):
|
||||
// CHECK: %[[VAR7:.*]] = torch.aten.mul.int %arg2, %[[INT2]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR8:.*]] = torch.aten.add.int %[[VAR7]], %[[INT7]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR9:.*]] = torch.prim.min.int %[[VAR8]], %[[INT68]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR10:.*]] = torch.aten.sub.int %[[VAR9]], %[[VAR7]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR11:.*]] = torch.aten.sub.int %[[INT7]], %[[VAR10]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR12:.*]] = torch.aten.add.int %[[VAR7]], %[[VAR10]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR13:.*]] = torch.aten.slice.Tensor %arg0, %[[INTM1]], %[[VAR7]], %[[VAR12]], %[[INT1]] : !torch.vtensor<[2,68],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[2,?],f32>
|
||||
// CHECK: %[[VAR14:.*]] = torch.prim.ListConstruct %[[INT0]], %[[VAR11]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR15:.*]] = torch.aten.constant_pad_nd %[[VAR13]], %[[VAR14]], %float0.000000e00 : !torch.vtensor<[2,?],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[2,?],f32>
|
||||
// CHECK: %[[VAR16:.*]] = torch.tensor_static_info_cast %[[VAR15]] : !torch.vtensor<[2,?],f32> to !torch.vtensor<[2,7],f32>
|
||||
// CHECK: %[[VAR17:.*]] = torch.aten.mul.Tensor %[[VAR16]], %[[VAR3]] : !torch.vtensor<[2,7],f32>, !torch.vtensor<[1,7],f32> -> !torch.vtensor<[2,7],f32>
|
||||
// CHECK: %[[VAR18:.*]] = torch.aten.fft_fft %[[VAR17]], %[[NONE]], %[[INTM1]], %[[NONE]] : !torch.vtensor<[2,7],f32>, !torch.none, !torch.int, !torch.none -> !torch.vtensor<[2,4],complex<f32>>
|
||||
// CHECK: %[[VAR19:.*]] = torch.aten.unsqueeze %[[VAR18]], %[[INTM1]] : !torch.vtensor<[2,4],complex<f32>>, !torch.int -> !torch.vtensor<[2,4,1],complex<f32>>
|
||||
// CHECK: %[[VAR20:.*]] = torch.aten.slice_scatter %arg3, %[[VAR19]], %[[INTM1]], %arg2, %[[NONE]], %[[INT1]] : !torch.vtensor<[2,4,31],complex<f32>>, !torch.vtensor<[2,4,1],complex<f32>>, !torch.int, !torch.int, !torch.none, !torch.int -> !torch.vtensor<[2,4,31],complex<f32>>
|
||||
// CHECK: torch.prim.Loop.condition %[[TRUE]], iter(%[[VAR20]] : !torch.vtensor<[2,4,31],complex<f32>>)
|
||||
// CHECK: } : (!torch.int, !torch.bool, !torch.vtensor<[2,4,31],complex<f32>>) -> !torch.vtensor<[2,4,31],complex<f32>>
|
||||
// CHECK: return %[[VAR6]] : !torch.vtensor<[2,4,31],complex<f32>>
|
||||
func.func @torch.aten.stft.center_2D_window_pad_left(%arg0: !torch.vtensor<[2,68],f32>, %arg1: !torch.vtensor<[6],f32>) -> !torch.vtensor<[2,4,31],complex<f32>> {
|
||||
%padmode = torch.constant.str "reflect"
|
||||
%nfft = torch.constant.int 7
|
||||
%hoplen = torch.constant.int 2
|
||||
%winlen = torch.constant.int 6
|
||||
%cstfalse = torch.constant.bool false
|
||||
%csttrue = torch.constant.bool true
|
||||
%0 = torch.aten.stft.center %arg0, %nfft, %hoplen, %winlen, %arg1, %cstfalse, %padmode, %cstfalse, %cstfalse, %csttrue : !torch.vtensor<[2,68],f32>, !torch.int, !torch.int, !torch.int, !torch.vtensor<[6],f32>, !torch.bool, !torch.str, !torch.bool, !torch.bool, !torch.bool -> !torch.vtensor<[2,4,31],complex<f32>>
|
||||
return %0 : !torch.vtensor<[2,4,31],complex<f32>>
|
||||
}
|
||||
|
||||
|
||||
// -----
|
||||
|
||||
|
||||
// CHECK-LABEL: func.func @torch.aten.stft.center_2D_hop_length_3_window_pad_both(
|
||||
// CHECK-SAME: %arg0: !torch.vtensor<[3,90],f32>, %arg1: !torch.vtensor<[8],f32>) -> !torch.vtensor<[3,6,27],complex<f32>> {
|
||||
// CHECK-DAG: %[[TRUE:.*]] = torch.constant.bool true
|
||||
// CHECK-DAG: %[[INT9:.*]] = torch.constant.int 9
|
||||
// CHECK-DAG: %[[INT6:.*]] = torch.constant.int 6
|
||||
// CHECK-DAG: %[[INT10:.*]] = torch.constant.int 10
|
||||
// CHECK-DAG: %[[INT3:.*]] = torch.constant.int 3
|
||||
// CHECK-DAG: %[[INT0:.*]] = torch.constant.int 0
|
||||
// CHECK-DAG: %[[INTM1:.*]] = torch.constant.int -1
|
||||
// CHECK-DAG: %[[INT1:.*]] = torch.constant.int 1
|
||||
// CHECK-DAG: %[[NONE:.*]] = torch.constant.none
|
||||
// CHECK-DAG: %float0.000000e00 = torch.constant.float 0.000000e+00
|
||||
// CHECK-DAG: %[[INT90:.*]] = torch.constant.int 90
|
||||
// CHECK-DAG: %[[INT27:.*]] = torch.constant.int 27
|
||||
// CHECK: %[[VAR0:.*]] = torch.prim.ListConstruct %[[INT1]], %[[INT1]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR1:.*]] = torch.aten.constant_pad_nd %arg1, %[[VAR0]], %float0.000000e00 : !torch.vtensor<[8],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[10],f32>
|
||||
// CHECK: %[[VAR2:.*]] = torch.prim.ListConstruct %[[INT1]], %[[INT10]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR3:.*]] = torch.aten.view %[[VAR1]], %[[VAR2]] : !torch.vtensor<[10],f32>, !torch.list<int> -> !torch.vtensor<[1,10],f32>
|
||||
// CHECK: %[[VAR4:.*]] = torch.prim.ListConstruct %[[INT3]], %[[INT6]], %[[INT27]] : (!torch.int, !torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR5:.*]] = torch.aten.empty.memory_format %[[VAR4]], %[[INT9]], %[[NONE]], %[[NONE]], %[[NONE]], %[[NONE]] : !torch.list<int>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[3,6,27],complex<f32>>
|
||||
// CHECK: %[[VAR6:.*]] = torch.prim.Loop %[[INT27]], %[[TRUE]], init(%[[VAR5]]) {
|
||||
// CHECK: ^bb0(%arg2: !torch.int, %arg3: !torch.vtensor<[3,6,27],complex<f32>>):
|
||||
// CHECK: %[[VAR7:.*]] = torch.aten.mul.int %arg2, %[[INT3]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR8:.*]] = torch.aten.add.int %[[VAR7]], %[[INT10]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR9:.*]] = torch.prim.min.int %[[VAR8]], %[[INT90]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR10:.*]] = torch.aten.sub.int %[[VAR9]], %[[VAR7]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR11:.*]] = torch.aten.sub.int %[[INT10]], %[[VAR10]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR12:.*]] = torch.aten.add.int %[[VAR7]], %[[VAR10]] : !torch.int, !torch.int -> !torch.int
|
||||
// CHECK: %[[VAR13:.*]] = torch.aten.slice.Tensor %arg0, %[[INTM1]], %[[VAR7]], %[[VAR12]], %[[INT1]] : !torch.vtensor<[3,90],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[3,?],f32>
|
||||
// CHECK: %[[VAR14:.*]] = torch.prim.ListConstruct %[[INT0]], %[[VAR11]] : (!torch.int, !torch.int) -> !torch.list<int>
|
||||
// CHECK: %[[VAR15:.*]] = torch.aten.constant_pad_nd %[[VAR13]], %[[VAR14]], %float0.000000e00 : !torch.vtensor<[3,?],f32>, !torch.list<int>, !torch.float -> !torch.vtensor<[3,?],f32>
|
||||
// CHECK: %[[VAR16:.*]] = torch.tensor_static_info_cast %[[VAR15]] : !torch.vtensor<[3,?],f32> to !torch.vtensor<[3,10],f32>
|
||||
// CHECK: %[[VAR17:.*]] = torch.aten.mul.Tensor %[[VAR16]], %[[VAR3]] : !torch.vtensor<[3,10],f32>, !torch.vtensor<[1,10],f32> -> !torch.vtensor<[3,10],f32>
|
||||
// CHECK: %[[VAR18:.*]] = torch.aten.fft_fft %[[VAR17]], %[[NONE]], %[[INTM1]], %[[NONE]] : !torch.vtensor<[3,10],f32>, !torch.none, !torch.int, !torch.none -> !torch.vtensor<[3,6],complex<f32>>
|
||||
// CHECK: %[[VAR19:.*]] = torch.aten.unsqueeze %[[VAR18]], %[[INTM1]] : !torch.vtensor<[3,6],complex<f32>>, !torch.int -> !torch.vtensor<[3,6,1],complex<f32>>
|
||||
// CHECK: %[[VAR20:.*]] = torch.aten.slice_scatter %arg3, %[[VAR19]], %[[INTM1]], %arg2, %[[NONE]], %[[INT1]] : !torch.vtensor<[3,6,27],complex<f32>>, !torch.vtensor<[3,6,1],complex<f32>>, !torch.int, !torch.int, !torch.none, !torch.int -> !torch.vtensor<[3,6,27],complex<f32>>
|
||||
// CHECK: torch.prim.Loop.condition %[[TRUE]], iter(%[[VAR20]] : !torch.vtensor<[3,6,27],complex<f32>>)
|
||||
// CHECK: } : (!torch.int, !torch.bool, !torch.vtensor<[3,6,27],complex<f32>>) -> !torch.vtensor<[3,6,27],complex<f32>>
|
||||
// CHECK: return %[[VAR6]] : !torch.vtensor<[3,6,27],complex<f32>>
|
||||
func.func @torch.aten.stft.center_2D_hop_length_3_window_pad_both(%arg0: !torch.vtensor<[3,90],f32>, %arg1: !torch.vtensor<[8],f32>) -> !torch.vtensor<[3,6,27],complex<f32>> {
|
||||
%padmode = torch.constant.str "reflect"
|
||||
%nfft = torch.constant.int 10
|
||||
%hoplen = torch.constant.int 3
|
||||
%winlen = torch.constant.int 8
|
||||
%cstfalse = torch.constant.bool false
|
||||
%csttrue = torch.constant.bool true
|
||||
%0 = torch.aten.stft.center %arg0, %nfft, %hoplen, %winlen, %arg1, %cstfalse, %padmode, %cstfalse, %cstfalse, %csttrue : !torch.vtensor<[3,90],f32>, !torch.int, !torch.int, !torch.int, !torch.vtensor<[8],f32>, !torch.bool, !torch.str, !torch.bool, !torch.bool, !torch.bool -> !torch.vtensor<[3,6,27],complex<f32>>
|
||||
return %0 : !torch.vtensor<[3,6,27],complex<f32>>
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue