Browse Source

in-class algoim_spark_alloc test

master
gjj 10 months ago
parent
commit
3ec39861f8
  1. 14
      algoim/organizer/primitive.hpp

14
algoim/organizer/primitive.hpp

@ -139,7 +139,8 @@ class PowerTensor : public Primitive
{ {
public: public:
xarray<real, 3> tensor; xarray<real, 3> tensor;
SparkStack<real>* sparkStackPtr;
// SparkStack<real>* sparkStackPtr;
void print() override { std::cout << "Power" << std::endl; } void print() override { std::cout << "Power" << std::endl; }
@ -151,7 +152,8 @@ public:
{ {
tensor.ext_ = ext_; tensor.ext_ = ext_;
tensor.data_ = nullptr; tensor.data_ = nullptr;
sparkStackPtr = algoim_spark_alloc_heap(real, tensor); algoim_spark_alloc(real, tensor);
// sparkStackPtr = algoim_spark_alloc_heap(real, tensor);
xarrayInit(tensor); xarrayInit(tensor);
} }
@ -159,10 +161,10 @@ public:
~PowerTensor() ~PowerTensor()
{ {
if (sparkStackPtr) { // if (sparkStackPtr) {
algoim_spark_release_heap(sparkStackPtr); // algoim_spark_release_heap(sparkStackPtr);
sparkStackPtr = nullptr; // sparkStackPtr = nullptr;
} // }
} }
}; };

Loading…
Cancel
Save