blob: 559f816e2088fe19f091159d3409d4e3181c34b3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
/*
* Copyright (c) Facebook, Inc. and its affiliates.
* All rights reserved.
* This source code is licensed under the BSD-style license found in the
* LICENSE file in the root directory of this source tree.
*/
#pragma once
#include <cmath>
#include <vector>
namespace fbgemm2 {
/*
* @brief Check and validate the buffers for reference and FBGEMM result.
*/
template <typename T>
int compare_validate_buffers(
const T* ref,
const T* test,
int m,
int n,
int ld,
T atol);
/*
* @brief Check if all entries are zero or not.
* If any entry is non-zero, return True;
* otherwise, return False.
*/
template <typename T>
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)
*/
template <typename T>
void transpose_matrix(T* ref, int n, int k);
} // namespace fbgemm2
|