diff options
| author | Jianyu Huang <jianyuhuang@fb.com> | 2019-01-31 02:06:15 +0300 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-01-31 02:17:18 +0300 |
| commit | 03a8fa506ee0822e3ab045f8e3df902abcfd2e74 (patch) | |
| tree | 95a5b96dfad4707fd006a9e7e00fad9becd63c6d /include/fbgemm/FbgemmFP16.h | |
| parent | 79333308f5e2fc242727879dcd3de3536b6ffc39 (diff) | |
Add threading for FBGEMM FP16
Summary: Add threading support for FBGEMM FP16 routines.
Reviewed By: dskhudia, jacobkahn
Differential Revision: D13792341
fbshipit-source-id: eb31a11340ac9fd0ee9b4f570d161e7c7e6a7602
Diffstat (limited to 'include/fbgemm/FbgemmFP16.h')
| -rw-r--r-- | include/fbgemm/FbgemmFP16.h | 23 |
1 files changed, 6 insertions, 17 deletions
diff --git a/include/fbgemm/FbgemmFP16.h b/include/fbgemm/FbgemmFP16.h index bebeb70..96deb49 100644 --- a/include/fbgemm/FbgemmFP16.h +++ b/include/fbgemm/FbgemmFP16.h @@ -192,14 +192,9 @@ class PackedGemmMatrixFP16 { const float* A, const PackedGemmMatrixFP16& Bp, const float beta, - float* C); - friend void cblas_gemm_compute( - const matrix_op_t transa, - const int m, - const float* A, - const PackedGemmMatrixFP16& Bp, - const float beta, - float* C); + float* C, + int thread_id, + int num_threads); }; /** @@ -211,13 +206,7 @@ extern void cblas_gemm_compute( const float* A, const PackedGemmMatrixFP16& Bp, const float beta, - float* C); -extern void cblas_gemm_compute( - const matrix_op_t transa, - const int m, - const float* A, - const PackedGemmMatrixFP16& Bp, - const float beta, - float* C); - + float* C, + int thread_id = 0, + int num_threads = 1); }; // namespace fbgemm |