Commit 90d7096a authored by Tat Dat Duong's avatar Tat Dat Duong
Browse files

chore: add setSize method

parent 32c2da51

implementation/src/BTreeContainer/BTreeContainerCuda.cu

+2 −0
Original line number Diff line number Diff line
@@ -400,6 +400,8 @@ public: 
    cudaDeviceSynchronize();

  }



  void setSize(size_t size) { mMemory.setSize(size); }



  void init(TNL::Containers::ArrayView<KeyType, Device> keys,

            TNL::Containers::ArrayView<ValueType, Device> values) {

    TNL_ASSERT_EQ(keys.getSize(), values.getSize(), "Values and keys must be the same size");

implementation/src/BTreeContainer/BTreeContainerHost.hpp

+2 −0
Original line number Diff line number Diff line
@@ -43,6 +43,8 @@ public: 


  void init() { mRoot = Tree::init(mAlloc); }



  void setSize(size_t size) { mMemory.setSize(size); }



  void insert(TNL::Containers::Array<KeyType, Device> &keys,

              TNL::Containers::Array<ValueType, Device> &values) {

    TNL_ASSERT_EQ(keys.getSize(), values.getSize(), "Values and keys must be the same size");