Skip to content

Commit

Permalink
update tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@raphaelDkhn
raphaelDkhn committed Mar 25, 2024
1 parent e7438fa commit 0b8f19c
Show file tree
Hide file tree
Showing 42 changed files with 1,187 additions and 1,165 deletions.
20 changes: 10 additions & 10 deletions nodegen/node/less_equal.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ def default():

x = Tensor(Dtype.U32, x.shape, x.flatten())
y = Tensor(Dtype.U32, y.shape, y.flatten())
z = Tensor(Dtype.U32, z.shape, z.flatten())
z = Tensor(Dtype.I32, z.shape, z.flatten())

name = "less_equal_u32"
make_test([x, y], z, "input_0.less_equal(@input_1)", name)
Expand All @@ -25,7 +25,7 @@ def broadcast():

x = Tensor(Dtype.U32, x.shape, x.flatten())
y = Tensor(Dtype.U32, y.shape, y.flatten())
z = Tensor(Dtype.U32, z.shape, z.flatten())
z = Tensor(Dtype.I32, z.shape, z.flatten())

name = "less_equal_u32_broadcast"
make_test([x, y], z, "input_0.less_equal(@input_1)", name)
Expand All @@ -42,7 +42,7 @@ def default():

x = Tensor(Dtype.I32, x.shape, x.flatten())
y = Tensor(Dtype.I32, y.shape, y.flatten())
z = Tensor(Dtype.U32, z.shape, z.flatten())
z = Tensor(Dtype.I32, z.shape, z.flatten())

name = "less_equal_i32"
make_test([x, y], z, "input_0.less_equal(@input_1)", name)
Expand All @@ -54,7 +54,7 @@ def broadcast():

x = Tensor(Dtype.I32, x.shape, x.flatten())
y = Tensor(Dtype.I32, y.shape, y.flatten())
z = Tensor(Dtype.U32, z.shape, z.flatten())
z = Tensor(Dtype.I32, z.shape, z.flatten())

name = "less_equal_i32_broadcast"
make_test([x, y], z, "input_0.less_equal(@input_1)", name)
Expand All @@ -71,7 +71,7 @@ def default():

x = Tensor(Dtype.I8, x.shape, x.flatten())
y = Tensor(Dtype.I8, y.shape, y.flatten())
z = Tensor(Dtype.U32, z.shape, z.flatten())
z = Tensor(Dtype.I32, z.shape, z.flatten())

name = "less_equal_i8"
make_test([x, y], z, "input_0.less_equal(@input_1)", name)
Expand All @@ -83,7 +83,7 @@ def broadcast():

x = Tensor(Dtype.I8, x.shape, x.flatten())
y = Tensor(Dtype.I8, y.shape, y.flatten())
z = Tensor(Dtype.U32, z.shape, z.flatten())
z = Tensor(Dtype.I32, z.shape, z.flatten())

name = "less_equal_i8_broadcast"
make_test([x, y], z, "input_0.less_equal(@input_1)", name)
Expand All @@ -102,7 +102,7 @@ def default():
x.flatten(), FixedImpl.FP8x23))
y = Tensor(Dtype.FP8x23, y.shape, to_fp(
y.flatten(), FixedImpl.FP8x23))
z = Tensor(Dtype.U32, z.shape, z.flatten())
z = Tensor(Dtype.I32, z.shape, z.flatten())

name = "less_equal_fp8x23"
make_test([x, y], z, "input_0.less_equal(@input_1)", name)
Expand All @@ -116,7 +116,7 @@ def broadcast():
x.flatten(), FixedImpl.FP8x23))
y = Tensor(Dtype.FP8x23, y.shape, to_fp(
y.flatten(), FixedImpl.FP8x23))
z = Tensor(Dtype.U32, z.shape, z.flatten())
z = Tensor(Dtype.I32, z.shape, z.flatten())

name = "less_equal_fp8x23_broadcast"
make_test([x, y], z, "input_0.less_equal(@input_1)", name)
Expand All @@ -135,7 +135,7 @@ def default():
x.flatten(), FixedImpl.FP16x16))
y = Tensor(Dtype.FP16x16, y.shape, to_fp(
y.flatten(), FixedImpl.FP16x16))
z = Tensor(Dtype.U32, z.shape, z.flatten())
z = Tensor(Dtype.I32, z.shape, z.flatten())

name = "less_equal_fp16x16"
make_test([x, y], z, "input_0.less_equal(@input_1)", name)
Expand All @@ -149,7 +149,7 @@ def broadcast():
x.flatten(), FixedImpl.FP16x16))
y = Tensor(Dtype.FP16x16, y.shape, to_fp(
y.flatten(), FixedImpl.FP16x16))
z = Tensor(Dtype.U32, z.shape, z.flatten())
z = Tensor(Dtype.I32, z.shape, z.flatten())

name = "less_equal_fp16x16_broadcast"
make_test([x, y], z, "input_0.less_equal(@input_1)", name)
Expand Down
Loading

0 comments on commit 0b8f19c

Please sign in to comment.