From 3ec39861f81e99f5c1cf3be9c50700d89be6971f Mon Sep 17 00:00:00 2001 From: gjj 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& phi, const uvector& x) class PowerTensor : public Primitive { public: - xarray tensor; - SparkStack* sparkStackPtr; + xarray tensor; + + // SparkStack* sparkStackPtr; void print() override { std::cout << "Power" << std::endl; } @@ -149,9 +150,10 @@ public: PowerTensor(uvector 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; + // } } };