Browse Source

in-class algoim_spark_alloc test

master
gjj 7 months ago
parent
commit
3ec39861f8
  1. 20
      algoim/organizer/primitive.hpp

20
algoim/organizer/primitive.hpp

@ -138,8 +138,9 @@ real evalBernstein(const xarray<real, N>& phi, const uvector<real, N>& x)
class PowerTensor : public Primitive 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; }
@ -149,9 +150,10 @@ public:
PowerTensor(uvector<int, 3> ext_) PowerTensor(uvector<int, 3> ext_)
{ {
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