fix exp

gizatechxyz · Oct 22, 2023 · 62edd56 · 62edd56
1 parent 05dbea5
commit 62edd56
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/operators/nn/functional/softmax.cairo b/src/operators/nn/functional/softmax.cairo
@@ -37,7 +37,8 @@ fn softmaxWide<
     impl WDrop: Drop<W>,
     impl TFixed: FixedTrait<T, TMAG>,
     impl WFixed: FixedTrait<W, WMAG>,
-    impl TPrint: PrintTrait<T>
+    impl TPrint: PrintTrait<T>,
+    impl WPrint: PrintTrait<W>
 >(
     z: @Tensor<T>, axis: usize
 ) -> Tensor<T> {

diff --git a/src/operators/tensor/math/exp.cairo b/src/operators/tensor/math/exp.cairo
@@ -52,7 +52,8 @@ fn exp_upcast<
     impl WCopy: Copy<W>,
     impl WDrop: Drop<W>,
     impl TIntoW: Into<T, W>,
-    impl TPrint: PrintTrait<T>
+    impl TPrint: PrintTrait<T>,
+    impl WPrint: PrintTrait<W>
 >(
     mut self: Tensor<T>
 ) -> Tensor<W> {
@@ -61,7 +62,7 @@ fn exp_upcast<
     loop {
         match self.data.pop_front() {
             Option::Some(item) => {
-                (*item).print();
+                (TIntoW::into(*item)).print();
 
                 result.append((TIntoW::into(*item)).exp());
             },