From 3ec39861f81e99f5c1cf3be9c50700d89be6971f Mon Sep 17 00:00:00 2001
From: gjj <dtouch@foxmail.com>
Date: Wed, 21 Aug 2024 12:01:46 +0800
Subject: [PATCH] in-class algoim_spark_alloc test

---
 algoim/organizer/primitive.hpp | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/algoim/organizer/primitive.hpp b/algoim/organizer/primitive.hpp
index d8376cf..a92bc17 100644
--- a/algoim/organizer/primitive.hpp
+++ b/algoim/organizer/primitive.hpp
@@ -138,8 +138,9 @@ real evalBernstein(const xarray<real, N>& phi, const uvector<real, N>& x)
 class PowerTensor : public Primitive
 {
 public:
-    xarray<real, 3>   tensor;
-    SparkStack<real>* sparkStackPtr;
+    xarray<real, 3> tensor;
+
+    // SparkStack<real>* sparkStackPtr;
 
     void print() override { std::cout << "Power" << std::endl; }
 
@@ -149,9 +150,10 @@ public:
 
     PowerTensor(uvector<int, 3> ext_)
     {
-        tensor.ext_   = ext_;
-        tensor.data_  = nullptr;
-        sparkStackPtr = algoim_spark_alloc_heap(real, tensor);
+        tensor.ext_  = ext_;
+        tensor.data_ = nullptr;
+        algoim_spark_alloc(real, tensor);
+        // sparkStackPtr = algoim_spark_alloc_heap(real, tensor);
         xarrayInit(tensor);
     }
 
@@ -159,10 +161,10 @@ public:
 
     ~PowerTensor()
     {
-        if (sparkStackPtr) {
-            algoim_spark_release_heap(sparkStackPtr);
-            sparkStackPtr = nullptr;
-        }
+        // if (sparkStackPtr) {
+        //     algoim_spark_release_heap(sparkStackPtr);
+        //     sparkStackPtr = nullptr;
+        // }
     }
 };