From 4ae5a3831356030e7c6d3c41afa486ad703b8a14 Mon Sep 17 00:00:00 2001 From: raphaelDkhn Date: Thu, 21 Mar 2024 11:49:07 +0100 Subject: [PATCH] test and refactor equal --- .../operators/tensor/tensor.equal.md | 8 +-- .../framework/operators/tensor/tensor.less.md | 4 +- nodegen/node/equal.py | 20 +++---- nodegen/node/less.py | 20 +++---- src/operators/tensor/core.cairo | 10 ++-- .../tensor/implementations/tensor_bool.cairo | 2 +- .../implementations/tensor_complex64.cairo | 2 +- .../implementations/tensor_fp16x16.cairo | 2 +- .../implementations/tensor_fp16x16wide.cairo | 2 +- .../implementations/tensor_fp32x32.cairo | 2 +- .../implementations/tensor_fp64x64.cairo | 2 +- .../implementations/tensor_fp8x23.cairo | 2 +- .../implementations/tensor_fp8x23wide.cairo | 2 +- .../tensor/implementations/tensor_i32.cairo | 2 +- .../tensor/implementations/tensor_i8.cairo | 2 +- .../tensor/implementations/tensor_u32.cairo | 2 +- src/operators/tensor/math/equal.cairo | 10 ++-- tests/nodes/equal_fp16x16.cairo | 16 ++--- tests/nodes/equal_fp16x16/input_0.cairo | 24 ++++---- tests/nodes/equal_fp16x16/input_1.cairo | 32 +++++----- tests/nodes/equal_fp16x16/output_0.cairo | 58 +++++++++---------- tests/nodes/equal_fp16x16_broadcast.cairo | 16 ++--- .../equal_fp16x16_broadcast/input_0.cairo | 8 +-- .../equal_fp16x16_broadcast/input_1.cairo | 6 +- .../equal_fp16x16_broadcast/output_0.cairo | 12 ++-- tests/nodes/equal_fp8x23.cairo | 14 ++--- tests/nodes/equal_fp8x23/input_0.cairo | 30 +++++----- tests/nodes/equal_fp8x23/input_1.cairo | 28 ++++----- tests/nodes/equal_fp8x23/output_0.cairo | 58 +++++++++---------- tests/nodes/equal_fp8x23_broadcast.cairo | 14 ++--- .../equal_fp8x23_broadcast/input_0.cairo | 6 +- .../equal_fp8x23_broadcast/input_1.cairo | 6 +- .../equal_fp8x23_broadcast/output_0.cairo | 12 ++-- tests/nodes/equal_i32.cairo | 16 ++--- tests/nodes/equal_i32/input_0.cairo | 31 +++++----- tests/nodes/equal_i32/input_1.cairo | 31 +++++----- tests/nodes/equal_i32/output_0.cairo | 58 +++++++++---------- tests/nodes/equal_i32_broadcast.cairo | 16 ++--- tests/nodes/equal_i32_broadcast/input_0.cairo | 7 ++- tests/nodes/equal_i32_broadcast/input_1.cairo | 5 +- .../nodes/equal_i32_broadcast/output_0.cairo | 12 ++-- tests/nodes/equal_i8.cairo | 16 ++--- tests/nodes/equal_i8/input_0.cairo | 31 +++++----- tests/nodes/equal_i8/input_1.cairo | 35 +++++------ tests/nodes/equal_i8/output_0.cairo | 58 +++++++++---------- tests/nodes/equal_i8_broadcast.cairo | 16 ++--- tests/nodes/equal_i8_broadcast/input_0.cairo | 7 ++- tests/nodes/equal_i8_broadcast/input_1.cairo | 7 ++- tests/nodes/equal_i8_broadcast/output_0.cairo | 12 ++-- tests/nodes/equal_u32.cairo | 14 +++-- tests/nodes/equal_u32/input_0.cairo | 29 +++++----- tests/nodes/equal_u32/input_1.cairo | 29 +++++----- tests/nodes/equal_u32/output_0.cairo | 58 +++++++++---------- tests/nodes/equal_u32_broadcast.cairo | 14 +++-- tests/nodes/equal_u32_broadcast/input_0.cairo | 9 +-- tests/nodes/equal_u32_broadcast/input_1.cairo | 7 ++- .../nodes/equal_u32_broadcast/output_0.cairo | 12 ++-- tests/nodes/less_fp16x16.cairo | 14 ++--- tests/nodes/less_fp16x16/input_0.cairo | 26 ++++----- tests/nodes/less_fp16x16/input_1.cairo | 32 +++++----- tests/nodes/less_fp16x16/output_0.cairo | 58 +++++++++---------- tests/nodes/less_fp16x16_broadcast.cairo | 14 ++--- .../less_fp16x16_broadcast/input_0.cairo | 28 ++++----- .../less_fp16x16_broadcast/input_1.cairo | 6 +- .../less_fp16x16_broadcast/output_0.cairo | 58 +++++++++---------- tests/nodes/less_fp8x23.cairo | 12 ++-- tests/nodes/less_fp8x23/input_0.cairo | 28 ++++----- tests/nodes/less_fp8x23/input_1.cairo | 26 ++++----- tests/nodes/less_fp8x23/output_0.cairo | 58 +++++++++---------- tests/nodes/less_fp8x23_broadcast.cairo | 12 ++-- .../nodes/less_fp8x23_broadcast/input_0.cairo | 32 +++++----- .../nodes/less_fp8x23_broadcast/input_1.cairo | 6 +- .../less_fp8x23_broadcast/output_0.cairo | 58 +++++++++---------- tests/nodes/less_i32.cairo | 14 ++--- tests/nodes/less_i32/input_0.cairo | 27 ++++----- tests/nodes/less_i32/input_1.cairo | 29 +++++----- tests/nodes/less_i32/output_0.cairo | 58 +++++++++---------- tests/nodes/less_i32_broadcast.cairo | 14 ++--- tests/nodes/less_i32_broadcast/input_0.cairo | 31 +++++----- tests/nodes/less_i32_broadcast/input_1.cairo | 9 +-- tests/nodes/less_i32_broadcast/output_0.cairo | 58 +++++++++---------- tests/nodes/less_i8.cairo | 14 ++--- tests/nodes/less_i8/input_0.cairo | 31 +++++----- tests/nodes/less_i8/input_1.cairo | 27 ++++----- tests/nodes/less_i8/output_0.cairo | 58 +++++++++---------- tests/nodes/less_i8_broadcast.cairo | 14 ++--- tests/nodes/less_i8_broadcast/input_0.cairo | 27 ++++----- tests/nodes/less_i8_broadcast/input_1.cairo | 7 ++- tests/nodes/less_i8_broadcast/output_0.cairo | 58 +++++++++---------- tests/nodes/less_u32.cairo | 12 ++-- tests/nodes/less_u32/input_0.cairo | 35 +++++------ tests/nodes/less_u32/input_1.cairo | 33 ++++++----- tests/nodes/less_u32/output_0.cairo | 58 +++++++++---------- tests/nodes/less_u32_broadcast.cairo | 12 ++-- tests/nodes/less_u32_broadcast/input_0.cairo | 31 +++++----- tests/nodes/less_u32_broadcast/input_1.cairo | 7 ++- tests/nodes/less_u32_broadcast/output_0.cairo | 58 +++++++++---------- 97 files changed, 1093 insertions(+), 1061 deletions(-) diff --git a/docs/framework/operators/tensor/tensor.equal.md b/docs/framework/operators/tensor/tensor.equal.md index 6e393c989..e8e91c20f 100644 --- a/docs/framework/operators/tensor/tensor.equal.md +++ b/docs/framework/operators/tensor/tensor.equal.md @@ -1,7 +1,7 @@ #tensor.equal ```rust - fn equal(self: @Tensor, other: @Tensor) -> Tensor; + fn equal(self: @Tensor, other: @Tensor) -> Tensor; ``` Check if two tensors are equal element-wise. @@ -20,7 +20,7 @@ The input tensors must have either: ## Returns -A new `Tensor` of booleans (1 if equal, 0 otherwise) with the same shape as the broadcasted inputs. +A new `Tensor` of booleans (1 if equal, 0 otherwise) with the same shape as the broadcasted inputs. ## Examples @@ -43,7 +43,7 @@ fn eq_example() -> Tensor { // We can call `equal` function as follows. return tensor_1.equal(@tensor_2); } ->>> [1,1,1,1,1,0,0,0] +>>> [true,true,true,true,true,false,false,false] ``` Case 2: Compare tensors with different shapes @@ -63,5 +63,5 @@ fn eq_example() -> Tensor { // We can call `equal` function as follows. return tensor_1.equal(@tensor_2); } ->>> [1,1,1,0,0,0,0,0,0] +>>> [true,true,true,false,false,false,false,false,false] ``` diff --git a/docs/framework/operators/tensor/tensor.less.md b/docs/framework/operators/tensor/tensor.less.md index d5d264d8a..797e51f89 100644 --- a/docs/framework/operators/tensor/tensor.less.md +++ b/docs/framework/operators/tensor/tensor.less.md @@ -53,7 +53,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor}; -fn less_example() -> Tensor { +fn less_example() -> Tensor { let tensor_1 = TensorTrait::::new( shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(), ); @@ -63,5 +63,5 @@ fn less_example() -> Tensor { // We can call `less` function as follows. return tensor_1.less(@tensor_2); } ->>> [0,0,0,0,0,0,0,1,1] +>>> [false,false,false,false,false,false,false,true,true] ``` diff --git a/nodegen/node/equal.py b/nodegen/node/equal.py index f995ae999..9474d3eca 100644 --- a/nodegen/node/equal.py +++ b/nodegen/node/equal.py @@ -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.BOOL, z.shape, z.flatten()) name = "equal_u32" make_test([x, y], z, "input_0.equal(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "equal_u32_broadcast" make_test([x, y], z, "input_0.equal(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "equal_i32" make_test([x, y], z, "input_0.equal(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "equal_i32_broadcast" make_test([x, y], z, "input_0.equal(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "equal_i8" make_test([x, y], z, "input_0.equal(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "equal_i8_broadcast" make_test([x, y], z, "input_0.equal(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "equal_fp8x23" make_test([x, y], z, "input_0.equal(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "equal_fp8x23_broadcast" make_test([x, y], z, "input_0.equal(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "equal_fp16x16" make_test([x, y], z, "input_0.equal(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "equal_fp16x16_broadcast" make_test([x, y], z, "input_0.equal(@input_1)", name) diff --git a/nodegen/node/less.py b/nodegen/node/less.py index 20b39263d..14af93201 100644 --- a/nodegen/node/less.py +++ b/nodegen/node/less.py @@ -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.BOOL, z.shape, z.flatten()) name = "less_u32" make_test([x, y], z, "input_0.less(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "less_u32_broadcast" make_test([x, y], z, "input_0.less(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "less_i32" make_test([x, y], z, "input_0.less(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "less_i32_broadcast" make_test([x, y], z, "input_0.less(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "less_i8" make_test([x, y], z, "input_0.less(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "less_i8_broadcast" make_test([x, y], z, "input_0.less(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "less_fp8x23" make_test([x, y], z, "input_0.less(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "less_fp8x23_broadcast" make_test([x, y], z, "input_0.less(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "less_fp16x16" make_test([x, y], z, "input_0.less(@input_1)", name) @@ -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.BOOL, z.shape, z.flatten()) name = "less_fp16x16_broadcast" make_test([x, y], z, "input_0.less(@input_1)", name) diff --git a/src/operators/tensor/core.cairo b/src/operators/tensor/core.cairo index 06791b024..e3dbc6e06 100644 --- a/src/operators/tensor/core.cairo +++ b/src/operators/tensor/core.cairo @@ -1085,7 +1085,7 @@ trait TensorTrait { /// #tensor.equal /// /// ```rust - /// fn equal(self: @Tensor, other: @Tensor) -> Tensor; + /// fn equal(self: @Tensor, other: @Tensor) -> Tensor; /// ``` /// /// Check if two tensors are equal element-wise. @@ -1104,7 +1104,7 @@ trait TensorTrait { /// /// ## Returns /// - /// A new `Tensor` of booleans (1 if equal, 0 otherwise) with the same shape as the broadcasted inputs. + /// A new `Tensor` of booleans (1 if equal, 0 otherwise) with the same shape as the broadcasted inputs. /// /// ## Examples /// @@ -1127,7 +1127,7 @@ trait TensorTrait { /// // We can call `equal` function as follows. /// return tensor_1.equal(@tensor_2); /// } - /// >>> [1,1,1,1,1,0,0,0] + /// >>> [true,true,true,true,true,false,false,false] /// ``` /// /// Case 2: Compare tensors with different shapes @@ -1147,10 +1147,10 @@ trait TensorTrait { /// // We can call `equal` function as follows. /// return tensor_1.equal(@tensor_2); /// } - /// >>> [1,1,1,0,0,0,0,0,0] + /// >>> [true,true,true,false,false,false,false,false,false] /// ``` /// - fn equal(self: @Tensor, other: @Tensor) -> Tensor; + fn equal(self: @Tensor, other: @Tensor) -> Tensor; /// #tensor.greater /// /// ```rust diff --git a/src/operators/tensor/implementations/tensor_bool.cairo b/src/operators/tensor/implementations/tensor_bool.cairo index 4d742454e..110cbb20e 100644 --- a/src/operators/tensor/implementations/tensor_bool.cairo +++ b/src/operators/tensor/implementations/tensor_bool.cairo @@ -109,7 +109,7 @@ impl BoolTensor of TensorTrait { panic(array!['not supported!']) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/implementations/tensor_complex64.cairo b/src/operators/tensor/implementations/tensor_complex64.cairo index 229a955a5..6870de2e2 100644 --- a/src/operators/tensor/implementations/tensor_complex64.cairo +++ b/src/operators/tensor/implementations/tensor_complex64.cairo @@ -128,7 +128,7 @@ impl Complex64Tensor of TensorTrait { math::log::log(*self) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/implementations/tensor_fp16x16.cairo b/src/operators/tensor/implementations/tensor_fp16x16.cairo index 64bef364c..6a22c9a6b 100644 --- a/src/operators/tensor/implementations/tensor_fp16x16.cairo +++ b/src/operators/tensor/implementations/tensor_fp16x16.cairo @@ -122,7 +122,7 @@ impl FP16x16Tensor of TensorTrait { math::log::log(*self) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/implementations/tensor_fp16x16wide.cairo b/src/operators/tensor/implementations/tensor_fp16x16wide.cairo index 8d45db74f..ddca96466 100644 --- a/src/operators/tensor/implementations/tensor_fp16x16wide.cairo +++ b/src/operators/tensor/implementations/tensor_fp16x16wide.cairo @@ -132,7 +132,7 @@ impl FP16x16WTensor of TensorTrait { math::log::log(*self) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/implementations/tensor_fp32x32.cairo b/src/operators/tensor/implementations/tensor_fp32x32.cairo index 58f860650..72d629224 100644 --- a/src/operators/tensor/implementations/tensor_fp32x32.cairo +++ b/src/operators/tensor/implementations/tensor_fp32x32.cairo @@ -120,7 +120,7 @@ impl FP32x32Tensor of TensorTrait { math::log::log(*self) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/implementations/tensor_fp64x64.cairo b/src/operators/tensor/implementations/tensor_fp64x64.cairo index 366a07e84..8d0b90f5e 100644 --- a/src/operators/tensor/implementations/tensor_fp64x64.cairo +++ b/src/operators/tensor/implementations/tensor_fp64x64.cairo @@ -119,7 +119,7 @@ impl FP64x64Tensor of TensorTrait { math::log::log(*self) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/implementations/tensor_fp8x23.cairo b/src/operators/tensor/implementations/tensor_fp8x23.cairo index f79c1b2b6..c319bb466 100644 --- a/src/operators/tensor/implementations/tensor_fp8x23.cairo +++ b/src/operators/tensor/implementations/tensor_fp8x23.cairo @@ -121,7 +121,7 @@ impl FP8x23Tensor of TensorTrait { math::log::log(*self) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/implementations/tensor_fp8x23wide.cairo b/src/operators/tensor/implementations/tensor_fp8x23wide.cairo index 849c7b4ca..c1b805677 100644 --- a/src/operators/tensor/implementations/tensor_fp8x23wide.cairo +++ b/src/operators/tensor/implementations/tensor_fp8x23wide.cairo @@ -124,7 +124,7 @@ impl FP8x23WTensor of TensorTrait { math::log::log(*self) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/implementations/tensor_i32.cairo b/src/operators/tensor/implementations/tensor_i32.cairo index fe4d0dcfb..42b7bb4db 100644 --- a/src/operators/tensor/implementations/tensor_i32.cairo +++ b/src/operators/tensor/implementations/tensor_i32.cairo @@ -117,7 +117,7 @@ impl I32Tensor of TensorTrait { panic(array!['not supported!']) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/implementations/tensor_i8.cairo b/src/operators/tensor/implementations/tensor_i8.cairo index 3e59a601a..53f6424b0 100644 --- a/src/operators/tensor/implementations/tensor_i8.cairo +++ b/src/operators/tensor/implementations/tensor_i8.cairo @@ -115,7 +115,7 @@ impl I8Tensor of TensorTrait { panic(array!['not supported!']) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/implementations/tensor_u32.cairo b/src/operators/tensor/implementations/tensor_u32.cairo index 23c49a40f..f9af20b65 100644 --- a/src/operators/tensor/implementations/tensor_u32.cairo +++ b/src/operators/tensor/implementations/tensor_u32.cairo @@ -114,7 +114,7 @@ impl U32Tensor of TensorTrait { panic(array!['not supported!']) } - fn equal(self: @Tensor, other: @Tensor) -> Tensor { + fn equal(self: @Tensor, other: @Tensor) -> Tensor { math::equal::equal(self, other) } diff --git a/src/operators/tensor/math/equal.cairo b/src/operators/tensor/math/equal.cairo index d2693acf9..7b2a1cb6c 100644 --- a/src/operators/tensor/math/equal.cairo +++ b/src/operators/tensor/math/equal.cairo @@ -6,15 +6,15 @@ use orion::operators::tensor::helpers::{ /// Cf: TensorTrait::equal docstring fn equal< T, - impl UsizeFTensor: TensorTrait, + impl BoolTensor: TensorTrait, impl TPartialEq: PartialEq, impl TCopy: Copy, impl TDrop: Drop >( y: @Tensor, z: @Tensor -) -> Tensor { +) -> Tensor { let broadcasted_shape = broadcast_shape(*y.shape, *z.shape); - let mut result: Array = array![]; + let mut result: Array = array![]; let num_elements = len_from_shape(broadcasted_shape); @@ -26,9 +26,9 @@ fn equal< let indices_other = broadcast_index_mapping(*z.shape, indices_broadcasted); if *(*y.data)[indices_self] == *(*z.data)[indices_other] { - result.append(1); + result.append(true); } else { - result.append(0); + result.append(false); } n += 1; diff --git a/tests/nodes/equal_fp16x16.cairo b/tests/nodes/equal_fp16x16.cairo index 38c3753cd..a71efdeda 100644 --- a/tests/nodes/equal_fp16x16.cairo +++ b/tests/nodes/equal_fp16x16.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorDiv}; -use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::FP16x16TensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::FP16x16TensorPartialEq; +use orion::operators::tensor::BoolTensor; +use orion::operators::tensor::BoolTensorPartialEq; +use orion::operators::tensor::{TensorTrait, Tensor}; #[test] #[available_gas(2000000000)] fn test_equal_fp16x16() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.equal(@input_1); + let y_0 = input_0.equal(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/equal_fp16x16/input_0.cairo b/tests/nodes/equal_fp16x16/input_0.cairo index 1c0bdb213..75bbd1b1a 100644 --- a/tests/nodes/equal_fp16x16/input_0.cairo +++ b/tests/nodes/equal_fp16x16/input_0.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorDiv}; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use orion::numbers::{FixedTrait, FP16x16}; fn input_0() -> Tensor { @@ -13,29 +13,29 @@ fn input_0() -> Tensor { data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 0, sign: false }); data.append(FP16x16 { mag: 196608, sign: true }); + data.append(FP16x16 { mag: 131072, sign: true }); data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 65536, sign: false }); - data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 65536, sign: true }); - data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 131072, sign: true }); data.append(FP16x16 { mag: 65536, sign: true }); - data.append(FP16x16 { mag: 65536, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp16x16/input_1.cairo b/tests/nodes/equal_fp16x16/input_1.cairo index c6e8fe0f6..76be7abd4 100644 --- a/tests/nodes/equal_fp16x16/input_1.cairo +++ b/tests/nodes/equal_fp16x16/input_1.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorDiv}; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use orion::numbers::{FixedTrait, FP16x16}; fn input_1() -> Tensor { @@ -10,32 +10,32 @@ fn input_1() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); + data.append(FP16x16 { mag: 196608, sign: true }); + data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 65536, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 65536, sign: true }); - data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 65536, sign: false }); + data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 131072, sign: true }); data.append(FP16x16 { mag: 196608, sign: true }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 196608, sign: true }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 65536, sign: false }); - data.append(FP16x16 { mag: 131072, sign: true }); data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 65536, sign: true }); - data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 65536, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 65536, sign: true }); - data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 196608, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp16x16/output_0.cairo b/tests/nodes/equal_fp16x16/output_0.cairo index 2078b6e18..5806c9c8f 100644 --- a/tests/nodes/equal_fp16x16/output_0.cairo +++ b/tests/nodes/equal_fp16x16/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp16x16_broadcast.cairo b/tests/nodes/equal_fp16x16_broadcast.cairo index 74eb5217e..1ca98f2ab 100644 --- a/tests/nodes/equal_fp16x16_broadcast.cairo +++ b/tests/nodes/equal_fp16x16_broadcast.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorDiv}; -use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::FP16x16TensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::FP16x16TensorPartialEq; +use orion::operators::tensor::BoolTensor; +use orion::operators::tensor::BoolTensorPartialEq; +use orion::operators::tensor::{TensorTrait, Tensor}; #[test] #[available_gas(2000000000)] fn test_equal_fp16x16_broadcast() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.equal(@input_1); + let y_0 = input_0.equal(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/equal_fp16x16_broadcast/input_0.cairo b/tests/nodes/equal_fp16x16_broadcast/input_0.cairo index a378b6d18..6e630308c 100644 --- a/tests/nodes/equal_fp16x16_broadcast/input_0.cairo +++ b/tests/nodes/equal_fp16x16_broadcast/input_0.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorDiv}; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use orion::numbers::{FixedTrait, FP16x16}; fn input_0() -> Tensor { @@ -9,9 +9,9 @@ fn input_0() -> Tensor { shape.append(2); let mut data = ArrayTrait::new(); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 65536, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp16x16_broadcast/input_1.cairo b/tests/nodes/equal_fp16x16_broadcast/input_1.cairo index 9a7b2b64d..b3280a1c0 100644 --- a/tests/nodes/equal_fp16x16_broadcast/input_1.cairo +++ b/tests/nodes/equal_fp16x16_broadcast/input_1.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorDiv}; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use orion::numbers::{FixedTrait, FP16x16}; fn input_1() -> Tensor { @@ -9,7 +9,7 @@ fn input_1() -> Tensor { shape.append(2); let mut data = ArrayTrait::new(); - data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 196608, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp16x16_broadcast/output_0.cairo b/tests/nodes/equal_fp16x16_broadcast/output_0.cairo index d2fab9fd0..e42f5e1e1 100644 --- a/tests/nodes/equal_fp16x16_broadcast/output_0.cairo +++ b/tests/nodes/equal_fp16x16_broadcast/output_0.cairo @@ -1,16 +1,16 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(2); shape.append(2); let mut data = ArrayTrait::new(); - data.append(1); - data.append(0); - data.append(0); - data.append(0); + data.append(false); + data.append(false); + data.append(false); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp8x23.cairo b/tests/nodes/equal_fp8x23.cairo index 112c71e8b..6a0d2aac1 100644 --- a/tests/nodes/equal_fp8x23.cairo +++ b/tests/nodes/equal_fp8x23.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::U32TensorPartialEq; use core::array::{ArrayTrait, SpanTrait}; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; +use orion::utils::{assert_eq, assert_seq_eq}; use orion::operators::tensor::FP8x23TensorPartialEq; +use orion::operators::tensor::BoolTensor; +use orion::operators::tensor::BoolTensorPartialEq; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorDiv}; -use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; #[test] #[available_gas(2000000000)] fn test_equal_fp8x23() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.equal(@input_1); + let y_0 = input_0.equal(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/equal_fp8x23/input_0.cairo b/tests/nodes/equal_fp8x23/input_0.cairo index 4f138ecb5..b13b78476 100644 --- a/tests/nodes/equal_fp8x23/input_0.cairo +++ b/tests/nodes/equal_fp8x23/input_0.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorDiv}; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; use orion::numbers::{FixedTrait, FP8x23}; fn input_0() -> Tensor { @@ -10,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: true }); + data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 25165824, sign: true }); + data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); - data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 8388608, sign: true }); data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: true }); data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 25165824, sign: true }); + data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 0, sign: false }); data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: true }); + data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 8388608, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp8x23/input_1.cairo b/tests/nodes/equal_fp8x23/input_1.cairo index b0010c344..4178c68b4 100644 --- a/tests/nodes/equal_fp8x23/input_1.cairo +++ b/tests/nodes/equal_fp8x23/input_1.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorDiv}; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; use orion::numbers::{FixedTrait, FP8x23}; fn input_1() -> Tensor { @@ -11,31 +11,31 @@ fn input_1() -> Tensor { let mut data = ArrayTrait::new(); data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: true }); data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 25165824, sign: true }); + data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: true }); + data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: true }); + data.append(FP8x23 { mag: 8388608, sign: false }); data.append(FP8x23 { mag: 8388608, sign: false }); data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: true }); + data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 0, sign: false }); data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: true }); + data.append(FP8x23 { mag: 0, sign: false }); data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 25165824, sign: true }); + data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); data.append(FP8x23 { mag: 0, sign: false }); data.append(FP8x23 { mag: 8388608, sign: true }); data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 25165824, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp8x23/output_0.cairo b/tests/nodes/equal_fp8x23/output_0.cairo index df58147cb..b38d8f0d7 100644 --- a/tests/nodes/equal_fp8x23/output_0.cairo +++ b/tests/nodes/equal_fp8x23/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp8x23_broadcast.cairo b/tests/nodes/equal_fp8x23_broadcast.cairo index 590193505..d40ab0cbc 100644 --- a/tests/nodes/equal_fp8x23_broadcast.cairo +++ b/tests/nodes/equal_fp8x23_broadcast.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::U32TensorPartialEq; use core::array::{ArrayTrait, SpanTrait}; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; +use orion::utils::{assert_eq, assert_seq_eq}; use orion::operators::tensor::FP8x23TensorPartialEq; +use orion::operators::tensor::BoolTensor; +use orion::operators::tensor::BoolTensorPartialEq; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorDiv}; -use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; #[test] #[available_gas(2000000000)] fn test_equal_fp8x23_broadcast() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.equal(@input_1); + let y_0 = input_0.equal(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/equal_fp8x23_broadcast/input_0.cairo b/tests/nodes/equal_fp8x23_broadcast/input_0.cairo index 31927729d..aa6318ba4 100644 --- a/tests/nodes/equal_fp8x23_broadcast/input_0.cairo +++ b/tests/nodes/equal_fp8x23_broadcast/input_0.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorDiv}; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; use orion::numbers::{FixedTrait, FP8x23}; fn input_0() -> Tensor { @@ -9,9 +9,9 @@ fn input_0() -> Tensor { shape.append(2); let mut data = ArrayTrait::new(); - data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 8388608, sign: true }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 25165824, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp8x23_broadcast/input_1.cairo b/tests/nodes/equal_fp8x23_broadcast/input_1.cairo index 6ffcceb3b..8c280300a 100644 --- a/tests/nodes/equal_fp8x23_broadcast/input_1.cairo +++ b/tests/nodes/equal_fp8x23_broadcast/input_1.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorDiv}; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; use orion::numbers::{FixedTrait, FP8x23}; fn input_1() -> Tensor { @@ -9,7 +9,7 @@ fn input_1() -> Tensor { shape.append(2); let mut data = ArrayTrait::new(); - data.append(FP8x23 { mag: 0, sign: false }); - data.append(FP8x23 { mag: 25165824, sign: true }); + data.append(FP8x23 { mag: 8388608, sign: true }); + data.append(FP8x23 { mag: 8388608, sign: false }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_fp8x23_broadcast/output_0.cairo b/tests/nodes/equal_fp8x23_broadcast/output_0.cairo index 417a71252..e42f5e1e1 100644 --- a/tests/nodes/equal_fp8x23_broadcast/output_0.cairo +++ b/tests/nodes/equal_fp8x23_broadcast/output_0.cairo @@ -1,16 +1,16 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(2); shape.append(2); let mut data = ArrayTrait::new(); - data.append(0); - data.append(0); - data.append(0); - data.append(0); + data.append(false); + data.append(false); + data.append(false); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i32.cairo b/tests/nodes/equal_i32.cairo index c2612bc6a..e9678d8b0 100644 --- a/tests/nodes/equal_i32.cairo +++ b/tests/nodes/equal_i32.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::I32TensorPartialEq; -use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::{TensorTrait, Tensor}; +use orion::operators::tensor::I32TensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{I32Tensor, I32TensorDiv}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensorPartialEq; +use orion::operators::tensor::{TensorTrait, Tensor}; #[test] #[available_gas(2000000000)] fn test_equal_i32() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.equal(@input_1); + let y_0 = input_0.equal(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/equal_i32/input_0.cairo b/tests/nodes/equal_i32/input_0.cairo index 0e2586cdb..1ddc0739f 100644 --- a/tests/nodes/equal_i32/input_0.cairo +++ b/tests/nodes/equal_i32/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I32Tensor, I32TensorDiv}; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(-3); - data.append(2); - data.append(2); - data.append(-3); - data.append(1); - data.append(-3); data.append(0); - data.append(-1); data.append(-3); data.append(-3); - data.append(-1); - data.append(-1); - data.append(0); data.append(1); - data.append(-1); - data.append(0); + data.append(-2); + data.append(2); + data.append(-2); data.append(0); data.append(-3); - data.append(1); data.append(-3); data.append(0); - data.append(1); data.append(-3); + data.append(-2); + data.append(2); + data.append(-2); + data.append(-2); + data.append(-2); data.append(-1); + data.append(0); data.append(2); + data.append(-1); data.append(-2); data.append(-3); + data.append(2); + data.append(0); + data.append(-3); + data.append(1); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i32/input_1.cairo b/tests/nodes/equal_i32/input_1.cairo index c6b5a6fbe..1721bc693 100644 --- a/tests/nodes/equal_i32/input_1.cairo +++ b/tests/nodes/equal_i32/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I32Tensor, I32TensorDiv}; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_1() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(-2); - data.append(-3); - data.append(-3); - data.append(-2); data.append(-3); + data.append(-1); + data.append(1); + data.append(0); + data.append(0); data.append(2); + data.append(-2); + data.append(-2); + data.append(-1); data.append(0); - data.append(1); data.append(-1); data.append(-3); - data.append(-1); data.append(2); + data.append(-2); data.append(-1); - data.append(1); data.append(-3); - data.append(-1); data.append(2); - data.append(2); - data.append(-2); + data.append(1); + data.append(1); data.append(-1); - data.append(-2); - data.append(-3); + data.append(2); + data.append(1); data.append(-1); data.append(-1); - data.append(0); - data.append(0); + data.append(-2); + data.append(2); data.append(-2); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i32/output_0.cairo b/tests/nodes/equal_i32/output_0.cairo index 0cdc2c69d..1e3bb5a60 100644 --- a/tests/nodes/equal_i32/output_0.cairo +++ b/tests/nodes/equal_i32/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i32_broadcast.cairo b/tests/nodes/equal_i32_broadcast.cairo index 012a7e165..8acfb1db1 100644 --- a/tests/nodes/equal_i32_broadcast.cairo +++ b/tests/nodes/equal_i32_broadcast.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::I32TensorPartialEq; -use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::{TensorTrait, Tensor}; +use orion::operators::tensor::I32TensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{I32Tensor, I32TensorDiv}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensorPartialEq; +use orion::operators::tensor::{TensorTrait, Tensor}; #[test] #[available_gas(2000000000)] fn test_equal_i32_broadcast() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.equal(@input_1); + let y_0 = input_0.equal(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/equal_i32_broadcast/input_0.cairo b/tests/nodes/equal_i32_broadcast/input_0.cairo index cefd51703..ad3cd4116 100644 --- a/tests/nodes/equal_i32_broadcast/input_0.cairo +++ b/tests/nodes/equal_i32_broadcast/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I32Tensor, I32TensorDiv}; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,8 +10,8 @@ fn input_0() -> Tensor { let mut data = ArrayTrait::new(); data.append(1); - data.append(-2); - data.append(0); + data.append(-3); data.append(1); + data.append(0); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i32_broadcast/input_1.cairo b/tests/nodes/equal_i32_broadcast/input_1.cairo index fed8199ca..8b33cf367 100644 --- a/tests/nodes/equal_i32_broadcast/input_1.cairo +++ b/tests/nodes/equal_i32_broadcast/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I32Tensor, I32TensorDiv}; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -8,7 +9,7 @@ fn input_1() -> Tensor { shape.append(2); let mut data = ArrayTrait::new(); - data.append(2); + data.append(1); data.append(1); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i32_broadcast/output_0.cairo b/tests/nodes/equal_i32_broadcast/output_0.cairo index 9a2391c78..75e094812 100644 --- a/tests/nodes/equal_i32_broadcast/output_0.cairo +++ b/tests/nodes/equal_i32_broadcast/output_0.cairo @@ -1,16 +1,16 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(2); shape.append(2); let mut data = ArrayTrait::new(); - data.append(0); - data.append(0); - data.append(0); - data.append(1); + data.append(true); + data.append(false); + data.append(true); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i8.cairo b/tests/nodes/equal_i8.cairo index e19689a8d..e5c0184fc 100644 --- a/tests/nodes/equal_i8.cairo +++ b/tests/nodes/equal_i8.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::U32TensorPartialEq; -use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::I8TensorPartialEq; -use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I8Tensor, I8TensorDiv}; +use orion::operators::tensor::BoolTensor; +use core::array::{ArrayTrait, SpanTrait}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensorPartialEq; +use orion::operators::tensor::{TensorTrait, Tensor}; #[test] #[available_gas(2000000000)] fn test_equal_i8() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.equal(@input_1); + let y_0 = input_0.equal(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/equal_i8/input_0.cairo b/tests/nodes/equal_i8/input_0.cairo index 09ea9171a..c89afb29b 100644 --- a/tests/nodes/equal_i8/input_0.cairo +++ b/tests/nodes/equal_i8/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I8Tensor, I8TensorDiv}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(-2); data.append(0); - data.append(-3); - data.append(-3); - data.append(1); - data.append(-3); data.append(-1); - data.append(1); data.append(-3); - data.append(2); data.append(0); + data.append(0); + data.append(-2); + data.append(2); data.append(-1); + data.append(0); + data.append(-2); + data.append(1); + data.append(-2); data.append(1); data.append(-3); + data.append(2); data.append(-3); data.append(-3); - data.append(1); - data.append(2); data.append(2); data.append(-3); - data.append(0); - data.append(0); - data.append(-1); - data.append(-2); data.append(-3); + data.append(-3); + data.append(-3); + data.append(2); + data.append(-2); + data.append(-1); + data.append(2); data.append(-1); - data.append(0); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i8/input_1.cairo b/tests/nodes/equal_i8/input_1.cairo index b1ab51213..c27187429 100644 --- a/tests/nodes/equal_i8/input_1.cairo +++ b/tests/nodes/equal_i8/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I8Tensor, I8TensorDiv}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_1() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(-1); - data.append(-1); - data.append(2); data.append(-3); - data.append(1); + data.append(0); data.append(-3); data.append(-2); - data.append(1); - data.append(0); - data.append(-1); - data.append(-1); - data.append(-1); - data.append(0); + data.append(-3); + data.append(2); + data.append(2); + data.append(-2); + data.append(2); + data.append(-2); data.append(-1); data.append(1); - data.append(1); + data.append(2); data.append(-1); - data.append(-2); data.append(-1); - data.append(2); - data.append(-3); data.append(1); + data.append(0); + data.append(2); + data.append(-2); data.append(-2); - data.append(1); data.append(-3); - data.append(1); data.append(-2); + data.append(2); + data.append(-1); + data.append(-2); + data.append(0); + data.append(-3); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i8/output_0.cairo b/tests/nodes/equal_i8/output_0.cairo index 452bd9d81..798787e19 100644 --- a/tests/nodes/equal_i8/output_0.cairo +++ b/tests/nodes/equal_i8/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i8_broadcast.cairo b/tests/nodes/equal_i8_broadcast.cairo index cc1fd18de..2713598b4 100644 --- a/tests/nodes/equal_i8_broadcast.cairo +++ b/tests/nodes/equal_i8_broadcast.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::U32TensorPartialEq; -use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::I8TensorPartialEq; -use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I8Tensor, I8TensorDiv}; +use orion::operators::tensor::BoolTensor; +use core::array::{ArrayTrait, SpanTrait}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensorPartialEq; +use orion::operators::tensor::{TensorTrait, Tensor}; #[test] #[available_gas(2000000000)] fn test_equal_i8_broadcast() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.equal(@input_1); + let y_0 = input_0.equal(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/equal_i8_broadcast/input_0.cairo b/tests/nodes/equal_i8_broadcast/input_0.cairo index 428f70adf..3f1c3803b 100644 --- a/tests/nodes/equal_i8_broadcast/input_0.cairo +++ b/tests/nodes/equal_i8_broadcast/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I8Tensor, I8TensorDiv}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -8,8 +9,8 @@ fn input_0() -> Tensor { shape.append(2); let mut data = ArrayTrait::new(); - data.append(-2); - data.append(2); + data.append(1); + data.append(1); data.append(-1); data.append(2); TensorTrait::new(shape.span(), data.span()) diff --git a/tests/nodes/equal_i8_broadcast/input_1.cairo b/tests/nodes/equal_i8_broadcast/input_1.cairo index 51830d474..e09cd290a 100644 --- a/tests/nodes/equal_i8_broadcast/input_1.cairo +++ b/tests/nodes/equal_i8_broadcast/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I8Tensor, I8TensorDiv}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -8,7 +9,7 @@ fn input_1() -> Tensor { shape.append(2); let mut data = ArrayTrait::new(); - data.append(-2); - data.append(-1); + data.append(0); + data.append(2); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_i8_broadcast/output_0.cairo b/tests/nodes/equal_i8_broadcast/output_0.cairo index d2fab9fd0..44d7202af 100644 --- a/tests/nodes/equal_i8_broadcast/output_0.cairo +++ b/tests/nodes/equal_i8_broadcast/output_0.cairo @@ -1,16 +1,16 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(2); shape.append(2); let mut data = ArrayTrait::new(); - data.append(1); - data.append(0); - data.append(0); - data.append(0); + data.append(false); + data.append(false); + data.append(false); + data.append(true); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_u32.cairo b/tests/nodes/equal_u32.cairo index 6591ecbc6..ef7a18f53 100644 --- a/tests/nodes/equal_u32.cairo +++ b/tests/nodes/equal_u32.cairo @@ -3,20 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::BoolTensorPartialEq; +use orion::operators::tensor::BoolTensor; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::{TensorTrait, Tensor}; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::{TensorTrait, Tensor}; #[test] #[available_gas(2000000000)] fn test_equal_u32() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.equal(@input_1); + let y_0 = input_0.equal(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/equal_u32/input_0.cairo b/tests/nodes/equal_u32/input_0.cairo index 9c3b88326..54f581c79 100644 --- a/tests/nodes/equal_u32/input_0.cairo +++ b/tests/nodes/equal_u32/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(1); data.append(5); - data.append(0); - data.append(2); data.append(5); + data.append(1); data.append(3); - data.append(2); data.append(5); + data.append(3); + data.append(2); + data.append(1); data.append(5); data.append(0); + data.append(4); + data.append(5); data.append(5); - data.append(0); - data.append(0); data.append(2); data.append(5); - data.append(4); - data.append(1); data.append(3); + data.append(4); data.append(1); - data.append(2); data.append(0); - data.append(5); data.append(2); - data.append(4); - data.append(2); - data.append(1); + data.append(5); + data.append(3); + data.append(3); + data.append(5); + data.append(5); + data.append(5); data.append(1); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_u32/input_1.cairo b/tests/nodes/equal_u32/input_1.cairo index 39331ddba..4545dcb02 100644 --- a/tests/nodes/equal_u32/input_1.cairo +++ b/tests/nodes/equal_u32/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -10,31 +11,31 @@ fn input_1() -> Tensor { let mut data = ArrayTrait::new(); data.append(4); - data.append(2); data.append(3); + data.append(5); + data.append(2); data.append(4); data.append(1); + data.append(5); data.append(2); data.append(0); - data.append(0); - data.append(4); - data.append(0); data.append(2); - data.append(0); + data.append(1); + data.append(4); + data.append(3); data.append(0); data.append(5); - data.append(3); + data.append(1); data.append(5); - data.append(2); - data.append(0); - data.append(2); - data.append(3); data.append(5); - data.append(4); + data.append(3); + data.append(0); data.append(1); - data.append(2); + data.append(3); + data.append(0); + data.append(3); data.append(5); data.append(3); - data.append(4); + data.append(2); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_u32/output_0.cairo b/tests/nodes/equal_u32/output_0.cairo index 339b2c840..573b6cac7 100644 --- a/tests/nodes/equal_u32/output_0.cairo +++ b/tests/nodes/equal_u32/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_u32_broadcast.cairo b/tests/nodes/equal_u32_broadcast.cairo index 30a7868a5..44d663e38 100644 --- a/tests/nodes/equal_u32_broadcast.cairo +++ b/tests/nodes/equal_u32_broadcast.cairo @@ -3,20 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::BoolTensorPartialEq; +use orion::operators::tensor::BoolTensor; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::{TensorTrait, Tensor}; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::{TensorTrait, Tensor}; #[test] #[available_gas(2000000000)] fn test_equal_u32_broadcast() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.equal(@input_1); + let y_0 = input_0.equal(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/equal_u32_broadcast/input_0.cairo b/tests/nodes/equal_u32_broadcast/input_0.cairo index d47d6b523..bb26dfe29 100644 --- a/tests/nodes/equal_u32_broadcast/input_0.cairo +++ b/tests/nodes/equal_u32_broadcast/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -8,9 +9,9 @@ fn input_0() -> Tensor { shape.append(2); let mut data = ArrayTrait::new(); - data.append(5); - data.append(5); - data.append(2); + data.append(3); data.append(0); + data.append(3); + data.append(1); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_u32_broadcast/input_1.cairo b/tests/nodes/equal_u32_broadcast/input_1.cairo index 7c4c61dff..39ab80b0c 100644 --- a/tests/nodes/equal_u32_broadcast/input_1.cairo +++ b/tests/nodes/equal_u32_broadcast/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -8,7 +9,7 @@ fn input_1() -> Tensor { shape.append(2); let mut data = ArrayTrait::new(); - data.append(4); - data.append(4); + data.append(3); + data.append(0); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/equal_u32_broadcast/output_0.cairo b/tests/nodes/equal_u32_broadcast/output_0.cairo index 417a71252..f757d7d01 100644 --- a/tests/nodes/equal_u32_broadcast/output_0.cairo +++ b/tests/nodes/equal_u32_broadcast/output_0.cairo @@ -1,16 +1,16 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorDiv}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(2); shape.append(2); let mut data = ArrayTrait::new(); - data.append(0); - data.append(0); - data.append(0); - data.append(0); + data.append(true); + data.append(true); + data.append(true); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp16x16.cairo b/tests/nodes/less_fp16x16.cairo index 04ac88b63..0f464463a 100644 --- a/tests/nodes/less_fp16x16.cairo +++ b/tests/nodes/less_fp16x16.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorSub}; +use orion::operators::tensor::BoolTensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{TensorTrait, Tensor}; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use orion::operators::tensor::FP16x16TensorPartialEq; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::{TensorTrait, Tensor}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::U32TensorPartialEq; #[test] #[available_gas(2000000000)] fn test_less_fp16x16() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.less(@input_1); + let y_0 = input_0.less(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/less_fp16x16/input_0.cairo b/tests/nodes/less_fp16x16/input_0.cairo index 41fa7524d..12c4be2fd 100644 --- a/tests/nodes/less_fp16x16/input_0.cairo +++ b/tests/nodes/less_fp16x16/input_0.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorSub}; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use orion::numbers::{FixedTrait, FP16x16}; fn input_0() -> Tensor { @@ -10,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(FP16x16 { mag: 65536, sign: true }); - data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 131072, sign: true }); data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 196608, sign: true }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 131072, sign: true }); data.append(FP16x16 { mag: 196608, sign: true }); + data.append(FP16x16 { mag: 131072, sign: true }); data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 65536, sign: true }); - data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 65536, sign: false }); - data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 0, sign: false }); data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 131072, sign: true }); data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 196608, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp16x16/input_1.cairo b/tests/nodes/less_fp16x16/input_1.cairo index fe0e56e41..8c6b3809a 100644 --- a/tests/nodes/less_fp16x16/input_1.cairo +++ b/tests/nodes/less_fp16x16/input_1.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorSub}; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use orion::numbers::{FixedTrait, FP16x16}; fn input_1() -> Tensor { @@ -10,32 +10,32 @@ fn input_1() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(FP16x16 { mag: 65536, sign: false }); - data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 65536, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 0, sign: false }); data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 65536, sign: true }); - data.append(FP16x16 { mag: 0, sign: false }); data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 65536, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 65536, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 196608, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp16x16/output_0.cairo b/tests/nodes/less_fp16x16/output_0.cairo index ff7a8e63d..19b525d8f 100644 --- a/tests/nodes/less_fp16x16/output_0.cairo +++ b/tests/nodes/less_fp16x16/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(0); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp16x16_broadcast.cairo b/tests/nodes/less_fp16x16_broadcast.cairo index 787c07448..750e149d7 100644 --- a/tests/nodes/less_fp16x16_broadcast.cairo +++ b/tests/nodes/less_fp16x16_broadcast.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorSub}; +use orion::operators::tensor::BoolTensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::{TensorTrait, Tensor}; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use orion::operators::tensor::FP16x16TensorPartialEq; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::{TensorTrait, Tensor}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::U32TensorPartialEq; #[test] #[available_gas(2000000000)] fn test_less_fp16x16_broadcast() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.less(@input_1); + let y_0 = input_0.less(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/less_fp16x16_broadcast/input_0.cairo b/tests/nodes/less_fp16x16_broadcast/input_0.cairo index 18782c0dd..cab41faaa 100644 --- a/tests/nodes/less_fp16x16_broadcast/input_0.cairo +++ b/tests/nodes/less_fp16x16_broadcast/input_0.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorSub}; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use orion::numbers::{FixedTrait, FP16x16}; fn input_0() -> Tensor { @@ -10,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 0, sign: false }); - data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 65536, sign: false }); + data.append(FP16x16 { mag: 0, sign: false }); data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 65536, sign: false }); - data.append(FP16x16 { mag: 131072, sign: true }); + data.append(FP16x16 { mag: 65536, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 0, sign: false }); data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 65536, sign: false }); - data.append(FP16x16 { mag: 65536, sign: true }); - data.append(FP16x16 { mag: 131072, sign: true }); - data.append(FP16x16 { mag: 0, sign: false }); data.append(FP16x16 { mag: 196608, sign: true }); - data.append(FP16x16 { mag: 131072, sign: false }); data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 196608, sign: true }); data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 65536, sign: true }); data.append(FP16x16 { mag: 196608, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp16x16_broadcast/input_1.cairo b/tests/nodes/less_fp16x16_broadcast/input_1.cairo index 743355c3d..65ffb99b7 100644 --- a/tests/nodes/less_fp16x16_broadcast/input_1.cairo +++ b/tests/nodes/less_fp16x16_broadcast/input_1.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorSub}; +use orion::operators::tensor::{FP16x16Tensor, FP16x16TensorAdd}; use orion::numbers::{FixedTrait, FP16x16}; fn input_1() -> Tensor { @@ -10,8 +10,8 @@ fn input_1() -> Tensor { shape.append(1); let mut data = ArrayTrait::new(); - data.append(FP16x16 { mag: 65536, sign: false }); data.append(FP16x16 { mag: 131072, sign: false }); - data.append(FP16x16 { mag: 0, sign: false }); + data.append(FP16x16 { mag: 131072, sign: false }); + data.append(FP16x16 { mag: 131072, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp16x16_broadcast/output_0.cairo b/tests/nodes/less_fp16x16_broadcast/output_0.cairo index 7d4613a88..b638d7ea1 100644 --- a/tests/nodes/less_fp16x16_broadcast/output_0.cairo +++ b/tests/nodes/less_fp16x16_broadcast/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(1); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(1); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(true); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp8x23.cairo b/tests/nodes/less_fp8x23.cairo index 6fe7b08b8..608b0e1e0 100644 --- a/tests/nodes/less_fp8x23.cairo +++ b/tests/nodes/less_fp8x23.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; +use orion::operators::tensor::BoolTensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; use orion::operators::tensor::FP8x23TensorPartialEq; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorSub}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::U32TensorPartialEq; #[test] #[available_gas(2000000000)] fn test_less_fp8x23() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.less(@input_1); + let y_0 = input_0.less(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/less_fp8x23/input_0.cairo b/tests/nodes/less_fp8x23/input_0.cairo index fbcd9f2a8..aee69b9c7 100644 --- a/tests/nodes/less_fp8x23/input_0.cairo +++ b/tests/nodes/less_fp8x23/input_0.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorSub}; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; use orion::numbers::{FixedTrait, FP8x23}; fn input_0() -> Tensor { @@ -10,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 0, sign: false }); data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 8388608, sign: false }); data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 0, sign: false }); data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: false }); data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 25165824, sign: true }); + data.append(FP8x23 { mag: 8388608, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 8388608, sign: true }); + data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: true }); + data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 25165824, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp8x23/input_1.cairo b/tests/nodes/less_fp8x23/input_1.cairo index e27ba84da..a013faaea 100644 --- a/tests/nodes/less_fp8x23/input_1.cairo +++ b/tests/nodes/less_fp8x23/input_1.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorSub}; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; use orion::numbers::{FixedTrait, FP8x23}; fn input_1() -> Tensor { @@ -10,32 +10,32 @@ fn input_1() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); + data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: true }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 8388608, sign: false }); data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 0, sign: false }); - data.append(FP8x23 { mag: 0, sign: false }); data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 0, sign: false }); - data.append(FP8x23 { mag: 8388608, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); - data.append(FP8x23 { mag: 8388608, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); - data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: true }); + data.append(FP8x23 { mag: 8388608, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp8x23/output_0.cairo b/tests/nodes/less_fp8x23/output_0.cairo index 33906ca90..180bf7d2e 100644 --- a/tests/nodes/less_fp8x23/output_0.cairo +++ b/tests/nodes/less_fp8x23/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(1); - data.append(0); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(true); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp8x23_broadcast.cairo b/tests/nodes/less_fp8x23_broadcast.cairo index e8b3155c5..19bdf417c 100644 --- a/tests/nodes/less_fp8x23_broadcast.cairo +++ b/tests/nodes/less_fp8x23_broadcast.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; +use orion::operators::tensor::BoolTensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; use orion::operators::tensor::FP8x23TensorPartialEq; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorSub}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::U32TensorPartialEq; #[test] #[available_gas(2000000000)] fn test_less_fp8x23_broadcast() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.less(@input_1); + let y_0 = input_0.less(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/less_fp8x23_broadcast/input_0.cairo b/tests/nodes/less_fp8x23_broadcast/input_0.cairo index 88ad7277b..425ca9e2e 100644 --- a/tests/nodes/less_fp8x23_broadcast/input_0.cairo +++ b/tests/nodes/less_fp8x23_broadcast/input_0.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorSub}; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; use orion::numbers::{FixedTrait, FP8x23}; fn input_0() -> Tensor { @@ -10,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 8388608, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); - data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 8388608, sign: true }); + data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); data.append(FP8x23 { mag: 16777216, sign: true }); data.append(FP8x23 { mag: 25165824, sign: true }); + data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: true }); + data.append(FP8x23 { mag: 25165824, sign: true }); + data.append(FP8x23 { mag: 8388608, sign: true }); + data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 0, sign: false }); - data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: true }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 8388608, sign: false }); data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: true }); - data.append(FP8x23 { mag: 25165824, sign: true }); data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 16777216, sign: false }); + data.append(FP8x23 { mag: 8388608, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: true }); + data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp8x23_broadcast/input_1.cairo b/tests/nodes/less_fp8x23_broadcast/input_1.cairo index 29b68e7c3..d7e956748 100644 --- a/tests/nodes/less_fp8x23_broadcast/input_1.cairo +++ b/tests/nodes/less_fp8x23_broadcast/input_1.cairo @@ -1,6 +1,6 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorSub}; +use orion::operators::tensor::{FP8x23Tensor, FP8x23TensorAdd}; use orion::numbers::{FixedTrait, FP8x23}; fn input_1() -> Tensor { @@ -10,8 +10,8 @@ fn input_1() -> Tensor { shape.append(1); let mut data = ArrayTrait::new(); - data.append(FP8x23 { mag: 0, sign: false }); + data.append(FP8x23 { mag: 16777216, sign: false }); data.append(FP8x23 { mag: 25165824, sign: true }); - data.append(FP8x23 { mag: 8388608, sign: true }); + data.append(FP8x23 { mag: 16777216, sign: true }); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_fp8x23_broadcast/output_0.cairo b/tests/nodes/less_fp8x23_broadcast/output_0.cairo index fbf242193..05547a4b4 100644 --- a/tests/nodes/less_fp8x23_broadcast/output_0.cairo +++ b/tests/nodes/less_fp8x23_broadcast/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(0); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i32.cairo b/tests/nodes/less_i32.cairo index 4a251b995..9c5cf2bbf 100644 --- a/tests/nodes/less_i32.cairo +++ b/tests/nodes/less_i32.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::{I32Tensor, I32TensorSub}; +use orion::operators::tensor::I32TensorPartialEq; +use orion::operators::tensor::BoolTensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::I32TensorPartialEq; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::U32TensorPartialEq; #[test] #[available_gas(2000000000)] fn test_less_i32() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.less(@input_1); + let y_0 = input_0.less(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/less_i32/input_0.cairo b/tests/nodes/less_i32/input_0.cairo index ab59d73f2..d89055483 100644 --- a/tests/nodes/less_i32/input_0.cairo +++ b/tests/nodes/less_i32/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I32Tensor, I32TensorSub}; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(-1); + data.append(1); + data.append(0); data.append(-2); data.append(2); - data.append(2); - data.append(2); - data.append(-2); + data.append(1); data.append(1); data.append(1); data.append(1); data.append(0); - data.append(-2); - data.append(-3); data.append(-1); data.append(-1); - data.append(-2); - data.append(2); - data.append(-2); - data.append(-1); - data.append(-3); data.append(0); data.append(0); data.append(-3); + data.append(1); + data.append(-1); + data.append(-3); data.append(-3); + data.append(2); + data.append(-1); + data.append(0); data.append(-2); data.append(2); + data.append(-1); + data.append(1); data.append(-3); - data.append(0); + data.append(-1); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i32/input_1.cairo b/tests/nodes/less_i32/input_1.cairo index cadacc785..a33bfe41b 100644 --- a/tests/nodes/less_i32/input_1.cairo +++ b/tests/nodes/less_i32/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I32Tensor, I32TensorSub}; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_1() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); + data.append(2); data.append(0); - data.append(-1); - data.append(0); - data.append(-2); + data.append(1); data.append(2); data.append(1); data.append(-1); + data.append(-2); + data.append(0); data.append(-3); - data.append(-3); + data.append(0); + data.append(1); + data.append(2); data.append(-2); data.append(-2); data.append(2); - data.append(-1); - data.append(-3); data.append(2); + data.append(2); + data.append(0); data.append(1); + data.append(-3); + data.append(-3); + data.append(-2); data.append(-2); - data.append(-1); data.append(-2); - data.append(1); data.append(-3); - data.append(-1); - data.append(0); - data.append(1); - data.append(0); + data.append(-3); data.append(2); - data.append(0); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i32/output_0.cairo b/tests/nodes/less_i32/output_0.cairo index bff093e8b..935f05c3b 100644 --- a/tests/nodes/less_i32/output_0.cairo +++ b/tests/nodes/less_i32/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(0); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i32_broadcast.cairo b/tests/nodes/less_i32_broadcast.cairo index 552150976..f82c428a8 100644 --- a/tests/nodes/less_i32_broadcast.cairo +++ b/tests/nodes/less_i32_broadcast.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; -use orion::operators::tensor::{I32Tensor, I32TensorSub}; +use orion::operators::tensor::I32TensorPartialEq; +use orion::operators::tensor::BoolTensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::I32TensorPartialEq; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::U32TensorPartialEq; #[test] #[available_gas(2000000000)] fn test_less_i32_broadcast() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.less(@input_1); + let y_0 = input_0.less(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/less_i32_broadcast/input_0.cairo b/tests/nodes/less_i32_broadcast/input_0.cairo index e49eaf809..bc2944af3 100644 --- a/tests/nodes/less_i32_broadcast/input_0.cairo +++ b/tests/nodes/less_i32_broadcast/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I32Tensor, I32TensorSub}; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(-3); + data.append(2); + data.append(1); data.append(-2); + data.append(2); data.append(1); + data.append(0); data.append(-3); - data.append(2); + data.append(-1); data.append(-1); data.append(-3); data.append(0); - data.append(2); - data.append(1); + data.append(-3); + data.append(-3); + data.append(-1); data.append(-1); - data.append(0); - data.append(1); data.append(1); + data.append(2); data.append(0); + data.append(-2); + data.append(-2); data.append(1); data.append(-3); - data.append(2); data.append(1); - data.append(2); - data.append(0); - data.append(0); - data.append(0); - data.append(-3); - data.append(0); - data.append(2); + data.append(-2); + data.append(-2); + data.append(-1); data.append(-2); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i32_broadcast/input_1.cairo b/tests/nodes/less_i32_broadcast/input_1.cairo index 99c50d44f..4f6d1b719 100644 --- a/tests/nodes/less_i32_broadcast/input_1.cairo +++ b/tests/nodes/less_i32_broadcast/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I32Tensor, I32TensorSub}; +use orion::operators::tensor::{I32Tensor, I32TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,8 +10,8 @@ fn input_1() -> Tensor { shape.append(1); let mut data = ArrayTrait::new(); - data.append(-3); - data.append(0); - data.append(1); + data.append(-2); + data.append(2); + data.append(-1); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i32_broadcast/output_0.cairo b/tests/nodes/less_i32_broadcast/output_0.cairo index 79967537c..3ffcf6eac 100644 --- a/tests/nodes/less_i32_broadcast/output_0.cairo +++ b/tests/nodes/less_i32_broadcast/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(1); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(true); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i8.cairo b/tests/nodes/less_i8.cairo index 085a6da35..00d6dddd4 100644 --- a/tests/nodes/less_i8.cairo +++ b/tests/nodes/less_i8.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; +use orion::operators::tensor::BoolTensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::I8TensorPartialEq; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::{I8Tensor, I8TensorSub}; -use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::I8TensorPartialEq; #[test] #[available_gas(2000000000)] fn test_less_i8() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.less(@input_1); + let y_0 = input_0.less(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/less_i8/input_0.cairo b/tests/nodes/less_i8/input_0.cairo index 28dd5a905..b2c30bed7 100644 --- a/tests/nodes/less_i8/input_0.cairo +++ b/tests/nodes/less_i8/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I8Tensor, I8TensorSub}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); + data.append(1); + data.append(-3); + data.append(1); + data.append(1); data.append(-3); data.append(0); data.append(-3); + data.append(0); + data.append(2); data.append(-2); - data.append(-1); data.append(2); - data.append(-1); - data.append(-3); - data.append(-1); - data.append(-3); - data.append(-3); - data.append(-1); - data.append(-1); data.append(-2); + data.append(0); data.append(-2); - data.append(-3); data.append(-1); - data.append(1); data.append(-2); - data.append(0); + data.append(-3); + data.append(-1); + data.append(-1); data.append(-3); data.append(2); + data.append(2); + data.append(1); data.append(-3); - data.append(-2); + data.append(1); data.append(2); - data.append(0); - data.append(0); + data.append(-1); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i8/input_1.cairo b/tests/nodes/less_i8/input_1.cairo index 9dc4e7a6c..cde86d6da 100644 --- a/tests/nodes/less_i8/input_1.cairo +++ b/tests/nodes/less_i8/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I8Tensor, I8TensorSub}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_1() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); + data.append(2); data.append(-3); - data.append(1); - data.append(-2); data.append(-3); - data.append(2); + data.append(-2); + data.append(1); data.append(-1); + data.append(-3); + data.append(0); data.append(0); + data.append(-2); + data.append(-1); data.append(-1); data.append(-2); - data.append(1); + data.append(0); + data.append(-3); data.append(0); data.append(2); - data.append(-2); + data.append(0); data.append(1); - data.append(-2); - data.append(-3); data.append(2); + data.append(-3); data.append(0); data.append(-1); data.append(0); - data.append(0); + data.append(-1); data.append(-2); - data.append(2); data.append(-1); - data.append(2); - data.append(0); - data.append(-3); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i8/output_0.cairo b/tests/nodes/less_i8/output_0.cairo index 9398cc8d3..0f911f707 100644 --- a/tests/nodes/less_i8/output_0.cairo +++ b/tests/nodes/less_i8/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(false); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i8_broadcast.cairo b/tests/nodes/less_i8_broadcast.cairo index fb705a81d..d100935c3 100644 --- a/tests/nodes/less_i8_broadcast.cairo +++ b/tests/nodes/less_i8_broadcast.cairo @@ -3,22 +3,22 @@ mod input_1; mod output_0; +use orion::operators::tensor::BoolTensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; -use orion::operators::tensor::I8TensorPartialEq; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::{I8Tensor, I8TensorSub}; -use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::I8TensorPartialEq; #[test] #[available_gas(2000000000)] fn test_less_i8_broadcast() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.less(@input_1); + let y_0 = input_0.less(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/less_i8_broadcast/input_0.cairo b/tests/nodes/less_i8_broadcast/input_0.cairo index eac6c02fd..19ab58b2d 100644 --- a/tests/nodes/less_i8_broadcast/input_0.cairo +++ b/tests/nodes/less_i8_broadcast/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I8Tensor, I8TensorSub}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(-2); - data.append(-2); - data.append(1); - data.append(1); - data.append(0); - data.append(0); data.append(-1); data.append(-3); data.append(0); data.append(2); + data.append(0); data.append(-3); data.append(1); - data.append(1); data.append(2); - data.append(-3); + data.append(0); data.append(2); data.append(-3); - data.append(2); + data.append(-1); + data.append(1); + data.append(-2); + data.append(-3); + data.append(-2); data.append(1); - data.append(0); data.append(-1); + data.append(2); data.append(0); data.append(-1); - data.append(1); + data.append(-2); data.append(0); + data.append(1); data.append(-2); - data.append(2); + data.append(-2); + data.append(1); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i8_broadcast/input_1.cairo b/tests/nodes/less_i8_broadcast/input_1.cairo index d593d06e4..88ed8e242 100644 --- a/tests/nodes/less_i8_broadcast/input_1.cairo +++ b/tests/nodes/less_i8_broadcast/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{I8Tensor, I8TensorSub}; +use orion::operators::tensor::{I8Tensor, I8TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,8 +10,8 @@ fn input_1() -> Tensor { shape.append(1); let mut data = ArrayTrait::new(); - data.append(-1); - data.append(-2); + data.append(1); data.append(-3); + data.append(1); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_i8_broadcast/output_0.cairo b/tests/nodes/less_i8_broadcast/output_0.cairo index f68e15ea8..ef4e9369d 100644 --- a/tests/nodes/less_i8_broadcast/output_0.cairo +++ b/tests/nodes/less_i8_broadcast/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(false); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_u32.cairo b/tests/nodes/less_u32.cairo index 412895527..b6546f1a2 100644 --- a/tests/nodes/less_u32.cairo +++ b/tests/nodes/less_u32.cairo @@ -3,20 +3,22 @@ mod input_1; mod output_0; +use orion::operators::tensor::BoolTensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; +use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::U32TensorPartialEq; #[test] #[available_gas(2000000000)] fn test_less_u32() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.less(@input_1); + let y_0 = input_0.less(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/less_u32/input_0.cairo b/tests/nodes/less_u32/input_0.cairo index 15b2924f3..89a0d8072 100644 --- a/tests/nodes/less_u32/input_0.cairo +++ b/tests/nodes/less_u32/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); + data.append(3); + data.append(3); + data.append(1); + data.append(3); + data.append(1); data.append(0); - data.append(2); data.append(0); - data.append(4); - data.append(5); + data.append(1); data.append(0); + data.append(3); + data.append(2); data.append(4); data.append(4); - data.append(1); data.append(5); data.append(3); - data.append(3); - data.append(2); - data.append(5); - data.append(5); - data.append(4); data.append(1); data.append(1); - data.append(4); - data.append(4); - data.append(2); - data.append(2); + data.append(5); + data.append(5); + data.append(0); + data.append(3); data.append(2); - data.append(4); - data.append(4); data.append(3); - data.append(5); + data.append(1); + data.append(3); + data.append(2); + data.append(1); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_u32/input_1.cairo b/tests/nodes/less_u32/input_1.cairo index e540f7f40..528c5927d 100644 --- a/tests/nodes/less_u32/input_1.cairo +++ b/tests/nodes/less_u32/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -10,31 +11,31 @@ fn input_1() -> Tensor { let mut data = ArrayTrait::new(); data.append(0); - data.append(1); - data.append(5); - data.append(5); + data.append(3); + data.append(4); data.append(2); - data.append(1); - data.append(1); data.append(2); - data.append(1); - data.append(3); - data.append(3); - data.append(0); data.append(5); - data.append(2); - data.append(0); - data.append(0); + data.append(4); + data.append(5); + data.append(4); data.append(0); data.append(4); - data.append(1); - data.append(1); + data.append(5); data.append(5); data.append(4); + data.append(0); data.append(4); data.append(2); + data.append(4); + data.append(5); + data.append(5); data.append(3); - data.append(2); + data.append(4); + data.append(5); + data.append(4); + data.append(5); + data.append(4); data.append(5); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_u32/output_0.cairo b/tests/nodes/less_u32/output_0.cairo index 75a278131..87600c456 100644 --- a/tests/nodes/less_u32/output_0.cairo +++ b/tests/nodes/less_u32/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(1); - data.append(1); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(true); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(true); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_u32_broadcast.cairo b/tests/nodes/less_u32_broadcast.cairo index 9a7ac7a22..d394bea89 100644 --- a/tests/nodes/less_u32_broadcast.cairo +++ b/tests/nodes/less_u32_broadcast.cairo @@ -3,20 +3,22 @@ mod input_1; mod output_0; +use orion::operators::tensor::BoolTensorPartialEq; use orion::utils::{assert_eq, assert_seq_eq}; +use orion::operators::tensor::U32TensorPartialEq; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; use core::array::{ArrayTrait, SpanTrait}; -use orion::operators::tensor::U32TensorPartialEq; #[test] #[available_gas(2000000000)] fn test_less_u32_broadcast() { let input_0 = input_0::input_0(); let input_1 = input_1::input_1(); - let z = output_0::output_0(); + let z_0 = output_0::output_0(); - let y = input_0.less(@input_1); + let y_0 = input_0.less(@input_1); - assert_eq(y, z); + assert_eq(y_0, z_0); } diff --git a/tests/nodes/less_u32_broadcast/input_0.cairo b/tests/nodes/less_u32_broadcast/input_0.cairo index 655814fc8..a740ac39b 100644 --- a/tests/nodes/less_u32_broadcast/input_0.cairo +++ b/tests/nodes/less_u32_broadcast/input_0.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; +use orion::numbers::NumberTrait; fn input_0() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,32 +10,32 @@ fn input_0() -> Tensor { shape.append(3); let mut data = ArrayTrait::new(); - data.append(5); - data.append(5); - data.append(3); - data.append(4); + data.append(1); data.append(4); + data.append(2); data.append(5); - data.append(3); - data.append(0); data.append(5); + data.append(0); data.append(3); data.append(3); data.append(0); - data.append(4); data.append(1); - data.append(5); - data.append(0); data.append(2); + data.append(4); + data.append(0); data.append(2); data.append(1); - data.append(3); - data.append(5); - data.append(5); + data.append(1); + data.append(2); data.append(5); + data.append(2); + data.append(3); data.append(4); - data.append(5); - data.append(5); + data.append(1); + data.append(3); + data.append(1); data.append(2); + data.append(5); + data.append(0); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_u32_broadcast/input_1.cairo b/tests/nodes/less_u32_broadcast/input_1.cairo index bcb20d101..f05f4343a 100644 --- a/tests/nodes/less_u32_broadcast/input_1.cairo +++ b/tests/nodes/less_u32_broadcast/input_1.cairo @@ -1,6 +1,7 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::{U32Tensor, U32TensorAdd}; +use orion::numbers::NumberTrait; fn input_1() -> Tensor { let mut shape = ArrayTrait::::new(); @@ -9,8 +10,8 @@ fn input_1() -> Tensor { shape.append(1); let mut data = ArrayTrait::new(); - data.append(3); - data.append(0); data.append(0); + data.append(5); + data.append(5); TensorTrait::new(shape.span(), data.span()) } diff --git a/tests/nodes/less_u32_broadcast/output_0.cairo b/tests/nodes/less_u32_broadcast/output_0.cairo index ad7acc0af..a9ade1a52 100644 --- a/tests/nodes/less_u32_broadcast/output_0.cairo +++ b/tests/nodes/less_u32_broadcast/output_0.cairo @@ -1,40 +1,40 @@ use core::array::{ArrayTrait, SpanTrait}; use orion::operators::tensor::{TensorTrait, Tensor}; -use orion::operators::tensor::{U32Tensor, U32TensorSub}; +use orion::operators::tensor::BoolTensor; -fn output_0() -> Tensor { +fn output_0() -> Tensor { let mut shape = ArrayTrait::::new(); shape.append(3); shape.append(3); shape.append(3); let mut data = ArrayTrait::new(); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(1); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); - data.append(0); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(false); + data.append(false); + data.append(false); + data.append(true); + data.append(true); + data.append(true); + data.append(true); + data.append(false); + data.append(true); TensorTrait::new(shape.span(), data.span()) }