diff options
author | Jongsoo Park <jongsoo@fb.com> | 2018-11-16 05:21:15 +0300 |
---|---|---|
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-11-16 05:24:41 +0300 |
commit | 8392eca198742b949529e18619a7ec9a25f4b399 (patch) | |
tree | da6f827a09b6a90d6294ca566f073ff387225a3b /test/TestUtils.h | |
parent | aa5b56a0de8b8f19047a9d797bf2d70cca4bc1f0 (diff) |
grouped (batched) gemm (#7)
Summary:
Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/7
This diff allows groups > 1 . Will have a separate diff for im2col + gemm fusion and conv with group > 1 .
Reviewed By: jianyuh
Differential Revision: D13039210
fbshipit-source-id: f7b3b0dbdb67fc6bc865de88292f034b252d029d
Diffstat (limited to 'test/TestUtils.h')
-rw-r--r-- | test/TestUtils.h | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/test/TestUtils.h b/test/TestUtils.h index 6cc365f..2cb7b88 100644 --- a/test/TestUtils.h +++ b/test/TestUtils.h @@ -32,9 +32,24 @@ bool check_all_zero_entries(const T* test, int m, int n); /* * @brief In-place transposition for nxk matrix ref. - * @params n number of rows in output (number of columns in input) - * @params k number of columns in output (number of rows in input) + * @params n number of rows in input (number of columns in output) + * @params k number of columns in input (number of rows in output) */ template <typename T> void transpose_matrix(T* ref, int n, int k); + +/* + * @brief Out-of-place transposition for M*N matrix ref. + * @params M number of rows in input + * @params K number of columns in input + */ +template <typename T> +void transpose_matrix( + int M, + int N, + const T* src, + int ld_src, + T* dst, + int ld_dst); + } // namespace fbgemm |