chunk_fusion.h

Go to the documentation of this file.

#pragma once
 
#include "fused/chunk_fusion/chunk_geometry.h"   // chunk geometry constants
#include "backend/types.h"
#include <cstddef>
#include <cstdint>
 
namespace fz {
class MemoryPool;
 
namespace fused {
 
enum class ChunkCoderKind { RZE, RRE };
 
size_t launchFusedChunkPfpl(
    ChunkCoderKind coder, const float* d_in, size_t n, float ebx2_r,
    uint32_t radius, float threshold, uint8_t* d_out, MemoryPool* pool, fz::stream_t stream,
    int chunk_bytes = 16384);
 
size_t launchFusedChunkPfplInverse(
    const uint8_t* d_archive, size_t archive_bytes, size_t output_bytes,
    float ebx2, bool inplace_outliers, uint32_t quant_radius,
    const float* d_outlier_vals, const uint32_t* d_outlier_idxs,
    uint32_t outlier_count, float* d_out, MemoryPool* pool, fz::stream_t stream,
    int chunk_bytes = 16384);
 
} // namespace fused
} // namespace fz