C++ wrappers for NVTX memory annotations

nvtxpp.md

namespace nvtx3{

enum class heap_kind{
   LINEAR,
   3D,
   ARRAY
};

struct heap_type{
   heap_type(heap_kind k){
     switch(k)
        case LINEAR: value_ = NVTX_MEM_TYPE_VIRTUAL_ADDRESS; break; 
        case 3D:     value_ = NVTX_MEM_TYPE_VIRTUAL_ADDRESS_3D; break;
        case ARRAY:  value_ = NVTX_MEM_TYPE_CUDA_ARRAY; break;
   }
   
  auto value(){ return value_; }
  
  private:
   uint32_t value_;
};

template <typename Domain>
struct suballocator{
   suballocator(heap_type type, /* TODO description */ ) :
      handle{nvtxMemHeapRegister(domain::get<Domain>(), NVTX_MEM_HEAP_SUB_ALLOCATOR, type.value(), /* TODO description */ } 
   }
   
   void register(void * p, std::size_t size){
      nvtxMemRegionRegister(domain::get<Domain>(), handle_, p, size);
   }
   
   void unregister(void * p){
      nvtxMemRegionUnregister(domain::get<Domain>(), p);
   }
   private:
      nvtxMemHeapHandle_t handle_;
};

} // namespace nvtx3