Commit 07d266ee authored by Tat Dat Duong's avatar Tat Dat Duong
Browse files

chore: use mask for host, code cleanup

parent acd4d3b1

implementation/BLinkTree/src/BNodeLatch/BNodeLatchCuda.cu

+2 −2
Original line number Diff line number Diff line
@@ -10,10 +10,10 @@ 


using namespace TNL;



template <typename Object, typename Device> struct Latch;

template <typename Object, typename Device> struct BNodeLatch;



template <typename KeyType, typename ValueType, size_t Order>

struct Latch<BNode<KeyType, ValueType, Order>, Devices::Cuda> {

struct BNodeLatch<BNode<KeyType, ValueType, Order>, Devices::Cuda> {

  using BTreeNode = BNode<KeyType, ValueType, Order>;



#ifdef HAVE_CUDA

implementation/BLinkTree/src/BNodeLatch/BNodeLatchHost.hpp

+2 −2
Original line number Diff line number Diff line
@@ -9,10 +9,10 @@ 


using namespace TNL;



template <typename Object, typename Device> struct Latch;

template <typename Object, typename Device> struct BNodeLatch;



template <typename KeyType, typename ValueType, size_t Order>

struct Latch<BNode<KeyType, ValueType, Order>, Devices::Host> {

struct BNodeLatch<BNode<KeyType, ValueType, Order>, Devices::Host> {

  using BTreeNode = BNode<KeyType, ValueType, Order>;



  void acquire(BTreeNode *node) {

implementation/BLinkTree/src/BNodeLatch/Default.hpp

+2 −2
Original line number Diff line number Diff line
@@ -3,10 +3,10 @@ 


using namespace TNL;



template <typename Object, typename Device> struct Latch {};

template <typename Object, typename Device> struct BNodeLatch {};



template <typename KeyType, typename ValueType, size_t Order, typename Device>

struct Latch<BNode<KeyType, ValueType, Order>, Device> {};

struct BNodeLatch<BNode<KeyType, ValueType, Order>, Device> {};



#include "./BNodeLatchHost.hpp"

#ifdef HAVE_CUDA

implementation/BLinkTree/src/BTree/BTreeCuda.cu

+1 −1
Original line number Diff line number Diff line
@@ -25,7 +25,7 @@ template <typename KeyType, typename ValueType, size_t Order> 
class BTree<BNode<KeyType, ValueType, Order>, Devices::Cuda> {

  using Device = Devices::Cuda;

  using BTreeNode = BNode<KeyType, ValueType, Order>;

  using BTreeLatch = Latch<BTreeNode, Device>;

  using BTreeLatch = BNodeLatch<BTreeNode, Device>;

  using BTreeAllocator = BumpAllocator<BTreeNode, Device>;

  using BTreeOperations = BNodeOperations<KeyType, ValueType, Order, Device>;

implementation/BLinkTree/src/BTree/BTreeHost.hpp

+1 −1
Original line number Diff line number Diff line
@@ -23,7 +23,7 @@ template <typename KeyType, typename ValueType, size_t Order> 
class BTree<BNode<KeyType, ValueType, Order>, Devices::Host> {

  using Device = Devices::Host;

  using BTreeNode = BNode<KeyType, ValueType, Order>;

  using BTreeLatch = Latch<BTreeNode, Device>;

  using BTreeLatch = BNodeLatch<BTreeNode, Device>;

  using BTreeAllocator = BumpAllocator<BTreeNode, Device>;

  using BTreeOperation = BNodeOperations<KeyType, ValueType, Order, Device>;