Skip to content

Commit

Permalink
minor, small comments
Browse files Browse the repository at this point in the history
  • Loading branch information
acharara committed Nov 15, 2017
1 parent e29f0ad commit 896f991
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 4 deletions.
2 changes: 1 addition & 1 deletion LICENSE
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
Copyright (c) 2017, King Abdullah University of Science and Technology
Copyright (c) 2012-, King Abdullah University of Science and Technology
All rights reserved.

Redistribution and use in source and binary forms, with or without
Expand Down
4 changes: 3 additions & 1 deletion src/batch_triangular/Xblas_core.ch
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,9 @@ void kblas_gemm_batch_wsquery(kblasHandle_t handle,
int B_row_off, int B_col_off,
int C_row_off, int C_col_off);

/// Non-strided batch gemm with offset
/**
* @brief Uniform-size batch non-strided GEMM with offset wrapper routine
*/
int kblas_gemm_batch( kblasHandle_t handle,
char transA, char transB,
const int m, const int n, const int k,
Expand Down
12 changes: 10 additions & 2 deletions src/batch_triangular/Xgemm_batch.cu
Original file line number Diff line number Diff line change
Expand Up @@ -42,10 +42,15 @@
//Non-Strided form

/**
* @brief Uniform-size batch non-strided GEMM with offset wrapper routine
*
* Workspace needed: device pointers
*
* @param[in] A_row_off row offset to sub-matrix of all A's
* @param[in] A_col_off column offset to sub-matrix of all A's
* @param[in] B_row_off row offset to sub-matrix of all B's
* @param[in] B_col_off column offset to sub-matrix of all B's
* @param[in] C_row_off row offset to sub-matrix of all C's
* @param[in] C_col_off column offset to sub-matrix of all C's
* @see kblasSgemm_batch() for details about rest of params.
* A, B, C: host pointer to array of device pointers to device buffers
*/
int kblas_gemm_batch( kblasHandle_t handle,
Expand All @@ -68,6 +73,7 @@ int kblas_gemm_batch( kblasHandle_t handle,
batchCount);
}

// Workspace needed: none
int kblas_gemm_batch( kblasHandle_t handle,
char transA, char transB,
const int m, const int n, const int k,
Expand All @@ -87,6 +93,8 @@ int kblas_gemm_batch( kblasHandle_t handle,
C_array, 0, 0, ldc,
batchCount);
}

// Workspace needed: none
extern "C"
int kblasXgemm_batch( kblasHandle_t handle,
char transA, char transB,
Expand Down

0 comments on commit 896f991

Please sign in to comment.