dump_buffer.cpp

#include <iostream>
#include <fstream>
#include <cstdlib>
#include <string>
#include <stdexcept>
#include <cassert>
#include <half/half.hpp>

using float16 = half_float::half;

template <typename T>
void dumpBufferToFile(const char* fileName, T* data, size_t dataNumItems)
{
    std::ofstream outFile(fileName, std::ios::binary);
    if(outFile)
    {
        outFile.write(reinterpret_cast<char*>(data), dataNumItems * sizeof(T));
        outFile.close();
        printf("Wrote output to file %s\n", fileName);
    }
    else
    {
        printf("Could not open file %s for writing\n", fileName);
    }
}

template <typename T>
void simpleBufferInit(T* data, size_t dataNumItems)
{
    const int dataOffset = 1;

    for(int i = 0; i < dataNumItems; i++)
        data[i] = static_cast<T>(i + dataOffset);
};

int main(int argc, char* argv[])
{
    if(argc != 4)
    {
        std::cerr << "Usage: " << argv[0] << " <fp64/fp32/fp16/int> <file> <number of elements> "
                  << std::endl;

        throw std::runtime_error("Invalid command argument!");
    };

    std::string precision(argv[1]);

    if(precision != "fp64" && precision != "fp32" && precision != "fp16" && precision != "int")
        throw std::runtime_error("Invalid precison of input data specified!");

    int numElements = atoi(argv[3]);

    assert(numElements > 0);

    if(precision == "fp64")
    {
        double* dataBuffer = new double[numElements];

        simpleBufferInit(dataBuffer, numElements);
        dumpBufferToFile(argv[2], dataBuffer, numElements);
    }
    else if(precision == "fp32")
    {
        float* dataBuffer = new float[numElements];

        simpleBufferInit(dataBuffer, numElements);
        dumpBufferToFile(argv[2], dataBuffer, numElements);
    }
    else if(precision == "fp16")
    {
        float16* dataBuffer = new float16[numElements];

        simpleBufferInit(dataBuffer, numElements);
        dumpBufferToFile(argv[2], dataBuffer, numElements);
        delete[] dataBuffer;
    }
    else
    {
        int* dataBuffer = new int[numElements];

        simpleBufferInit(dataBuffer, numElements);
        dumpBufferToFile(argv[2], dataBuffer, numElements);
        delete[] dataBuffer;
    };
};