GEMM Operations - GEMM Operations - 5.3 English - 57404

#include "aocl_dlp.h"

// Basic f32 GEMM call: C = alpha * A * B + beta * C
aocl_gemm_f32f32f32of32(
    'R',        // Storage format (R=row-major, C=column-major)
    'N',        // TransA (N=no transpose, T=transpose)
    'N',        // TransB
    m, n, k,    // Matrix dimensions
    1.0f,       // alpha scalar
    a, lda, 'N', // Matrix A, leading dimension, memory format
    b, ldb, 'N', // Matrix B, leading dimension, memory format
    0.0f,       // beta scalar
    c, ldc,     // Matrix C, leading dimension
    NULL        // Post-operations metadata (NULL = no post-ops)
);

// Get buffer size needed for reordering matrix B
msz_t buffer_size = aocl_get_reorder_buf_size_f32f32f32of32(
    'R',     // Storage order (row-major)
    'N',     // TransB
    'B',     // Matrix to reorder ('A' or 'B')
    k, n,    // Dimensions (rows and cols of B)
    NULL     // Post-operations metadata
);

// Allocate buffer and reorder matrix B
float* reordered_b = (float*)malloc(buffer_size);
aocl_reorder_f32f32f32of32(
    'R',           // Storage order
    'N',           // TransB
    'B',           // Matrix to reorder
    b,             // Source matrix
    reordered_b,   // Destination buffer
    k, n,          // Dimensions
    ldb,           // Leading dimension of source
    NULL           // Post-operations metadata
);

// Use reordered matrix in GEMM calls
aocl_gemm_f32f32f32of32(
    'R', 'N', 'N', m, n, k,
    1.0f, a, lda, 'N',
    reordered_b, ldb, 'R',  // 'R' indicates reordered format
    0.0f, c, ldc, NULL
);

// Clean up
free(reordered_b);

// Batch f32 GEMM example
aocl_batch_gemm_f32f32f32of32(
    order_array,                          // Array of storage layouts
    transa_array, transb_array,           // Arrays of transpose flags
    m_array, n_array, k_array,            // Arrays of dimensions
    alpha_array,                          // Array of alpha scalars
    (const float**)a_array, lda_array,    // Arrays of A matrices
    (const float**)b_array, ldb_array,    // Arrays of B matrices
    beta_array,                           // Array of beta scalars
    c_array, ldc_array,                   // Arrays of C matrices
    group_count, group_sizes,             // Grouping parameters
    mem_format_a_array, mem_format_b_array, // Memory format arrays
    post_ops_array                        // Array of post-operations metadata
);

float alpha=1.0f, beta=0.0f;
dlp_metadata_t meta = {0};
aocl_gemm_f32f32f32of32('R', 'N', 'N', m, n, k,
    alpha, a, lda, 'N', b, ldb, 'N', beta, c, ldc, &meta);

if (meta.error_hndl.error_code != DLP_CLSC_SUCCESS) {
    fprintf(stderr, "Error code: %d\n", meta.error_hndl.error_code);
}