Update examples to use kp::Memory instead of kp::Tensor as required.

Only done when creating a vector of shared_ptrs of Tensors as these can't be automatically converted to a vector of shared_ptrs of Memorys to be passed to Manager::algorithm
KomputeProject · Aug 19, 2024 · d6f3c91 · d6f3c91
1 parent 57b62ee
commit d6f3c91
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/android/android-simple/app/src/main/cpp/KomputeModelML.cpp b/examples/android/android-simple/app/src/main/cpp/KomputeModelML.cpp
@@ -38,7 +38,7 @@ KomputeModelML::train(std::vector<float> yData,
 
         std::shared_ptr<kp::TensorT<float>> lOut = mgr.tensor(zerosData);
 
-        std::vector<std::shared_ptr<kp::Tensor>> params = { xI,  xJ,    y,
+        std::vector<std::shared_ptr<kp::Memory>> params = { xI,  xJ,    y,
                                                             wIn, wOutI, wOutJ,
                                                             bIn, bOut,  lOut };
 

diff --git a/examples/array_multiplication/src/main.cpp b/examples/array_multiplication/src/main.cpp
@@ -18,7 +18,7 @@ main()
     std::shared_ptr<kp::TensorT<float>> tensorOut =
       mgr.tensor({ 0.0, 0.0, 0.0 });
 
-    const std::vector<std::shared_ptr<kp::Tensor>> params = { tensorInA,
+    const std::vector<std::shared_ptr<kp::Memory>> params = { tensorInA,
                                                               tensorInB,
                                                               tensorOut };
 

diff --git a/examples/logistic_regression/src/main.cpp b/examples/logistic_regression/src/main.cpp
@@ -28,7 +28,7 @@ main()
 
     std::shared_ptr<kp::TensorT<float>> lOut = mgr.tensor({ 0, 0, 0, 0, 0 });
 
-    std::vector<std::shared_ptr<kp::Tensor>> params = { xI,  xJ,    y,
+    std::vector<std::shared_ptr<kp::Memory>> params = { xI,  xJ,    y,
                                                         wIn, wOutI, wOutJ,
                                                         bIn, bOut,  lOut };