bitplane_rze_kernels.h

Go to the documentation of this file.

#pragma once
 
#include <cuda_runtime.h>
#include <cstddef>
#include <cstdint>
 
namespace fz {
namespace bitplane_rze {
 
enum : int {
  BPRZE_HDR_HEADER    = 0,
  BPRZE_HDR_BITFLAG   = 1,
  BPRZE_HDR_STARTPOS  = 2,
  BPRZE_HDR_BITSTREAM = 3,
  BPRZE_HDR_END       = 4,
};
 
constexpr size_t kBitplaneRzeHeaderBytes = 128;
 
struct ArchiveHeader {
  size_t   original_len;             
  uint32_t entry[BPRZE_HDR_END + 1]; 
  uint8_t  padding[kBitplaneRzeHeaderBytes - sizeof(size_t) -
                   sizeof(uint32_t) * (BPRZE_HDR_END + 1)];
};
static_assert(sizeof(ArchiveHeader) == kBitplaneRzeHeaderBytes,
              "ArchiveHeader must be exactly 128 bytes");
 
struct Config {
  size_t data_len;    
  size_t pad_len;     
  size_t data_bytes;  
  size_t chunk_size;  
  size_t grid_x;      
};
 
Config configure(size_t data_len);
 
size_t maxArchiveBytes(size_t data_len);
 
void launchEncode(
    const uint16_t* d_in, const Config& cfg,
    uint32_t* d_offset_counter, uint32_t* d_bitflag, uint32_t* d_start_pos,
    uint8_t* d_comp_out, uint32_t* d_comp_len, cudaStream_t stream);
 
void launchDecode(
    const uint8_t* d_bitstream, const uint32_t* d_bitflag,
    const uint32_t* d_start_pos, uint16_t* d_out, const Config& cfg,
    cudaStream_t stream);
 
}  // namespace bitplane_rze
}  // namespace fz