fused_block.h

Go to the documentation of this file.

#pragma once
 
#include "backend/types.h"
#include <cstddef>
#include <cstdint>
 
namespace fz {
class MemoryPool;
 
namespace fused {
 
size_t launchFusedCuszp2Compress(
    const float* d_in, size_t n, float abs_eb, uint32_t block_size, bool outlier,
    uint8_t* d_out, MemoryPool* pool, fz::stream_t stream);
 
size_t launchFusedCuszp3Compress(
    const float* d_in, size_t dx, size_t dy, float abs_eb, uint32_t tx, uint32_t ty,
    uint8_t* d_out, MemoryPool* pool, fz::stream_t stream);
 
} // namespace fused
} // namespace fz