#include <hebench_matmult.h>
|
static DataLoader::Ptr | create (std::uint64_t rows_a, std::uint64_t cols_a, std::uint64_t cols_b, std::uint64_t batch_size_mat_a, std::uint64_t batch_size_mat_b, hebench::APIBridge::DataType data_type) |
|
static DataLoader::Ptr | create (std::uint64_t rows_a, std::uint64_t cols_a, std::uint64_t cols_b, std::uint64_t expected_sample_size_mat_a, std::uint64_t expected_sample_size_mat_b, hebench::APIBridge::DataType data_type, const std::string &dataset_filename) |
|
static std::size_t | sizeOf (hebench::APIBridge::DataType data_type) |
|
static unique_ptr_custom_deleter< hebench::APIBridge::DataPackCollection > | createDataPackCollection (std::uint64_t data_pack_count) |
| Creates shallow packed data that self cleans up. More...
|
|
static unique_ptr_custom_deleter< hebench::APIBridge::DataPack > | createDataPack (std::uint64_t buffer_count, std::uint64_t param_position) |
| Creates shallow data pack that self cleans up. More...
|
|
static unique_ptr_custom_deleter< hebench::APIBridge::NativeDataBuffer > | createDataBuffer (std::uint64_t size, std::int64_t tag) |
|
|
void | computeResult (std::vector< hebench::APIBridge::NativeDataBuffer * > &result, const std::uint64_t *param_data_pack_indices, hebench::APIBridge::DataType data_type) override |
| Computes result of the operation on the input data given the of the input sample. More...
|
|
| DataLoaderCompute () |
|
| PartialDataLoader () |
|
void | init (hebench::APIBridge::DataType data_type, std::size_t input_dim, const std::size_t *input_sample_count_per_dim, const std::uint64_t *input_count_per_dim, std::size_t output_dim, const std::uint64_t *output_count_per_dim, bool allocate_output) |
| Initializes dimensions of inputs and outputs. No allocation is performed. More...
|
|
void | init (const std::string &filename, hebench::APIBridge::DataType data_type, std::size_t expected_input_dim, const std::size_t *max_input_sample_count_per_dim, const std::uint64_t *expected_input_count_per_dim, std::size_t expected_output_dim, const std::uint64_t *expected_output_count_per_dim) |
| Loads a dataset from a file. More...
|
|
std::vector< std::shared_ptr< hebench::APIBridge::DataPack > > | getResultTempDataPacks (std::uint64_t result_index) const |
| Retrieves a pre-allocated result providing memory space to store a single operation result sample. More...
|
|
std::vector< std::shared_ptr< hebench::APIBridge::DataPack > > | getResultTempDataPacks (const std::uint64_t *param_data_pack_indices) const |
| Retrieves a pre-allocated result providing memory space to store a single operation result sample. More...
|
|
std::vector< std::shared_ptr< hebench::APIBridge::DataPack > > | getResultTempDataPacks () const |
| Retrieves a pre-allocated result providing memory space to store a single operation result sample. More...
|
|
| IDataLoader () |
|
Definition at line 62 of file hebench_matmult.h.
◆ Ptr
◆ ~DataLoader()
hebench::TestHarness::MatrixMultiply::DataLoader::~DataLoader |
( |
| ) |
|
|
inlineoverride |
◆ computeResult()
Computes result of the operation on the input data given the of the input sample.
- Parameters
-
| result | Vector where to store the result. Vector comes pre-initialized. This method's task is to fill up the data buffers with the result values. |
[in] | param_data_pack_indices | For each operation parameter, this array indicates the sample index to use for the operation. Must contain, at least, getParameterCount() elements. |
[in] | data_type | Data type of the data pointed by the inputs and result. |
Implements hebench::TestHarness::DataLoaderCompute.
Definition at line 365 of file hebench_matmult.cpp.
◆ create() [1/2]
DataLoader::Ptr hebench::TestHarness::MatrixMultiply::DataLoader::create |
( |
std::uint64_t |
rows_a, |
|
|
std::uint64_t |
cols_a, |
|
|
std::uint64_t |
cols_b, |
|
|
std::uint64_t |
batch_size_mat_a, |
|
|
std::uint64_t |
batch_size_mat_b, |
|
|
hebench::APIBridge::DataType |
data_type |
|
) |
| |
|
static |
◆ create() [2/2]
DataLoader::Ptr hebench::TestHarness::MatrixMultiply::DataLoader::create |
( |
std::uint64_t |
rows_a, |
|
|
std::uint64_t |
cols_a, |
|
|
std::uint64_t |
cols_b, |
|
|
std::uint64_t |
expected_sample_size_mat_a, |
|
|
std::uint64_t |
expected_sample_size_mat_b, |
|
|
hebench::APIBridge::DataType |
data_type, |
|
|
const std::string & |
dataset_filename |
|
) |
| |
|
static |
The documentation for this class was generated from the following files: