Rename "HALF" and "sh" to "BFLOAT16" and "sb"

This commit is contained in:
Martin Kroeker 2020-10-11 23:43:36 +02:00 committed by GitHub
parent 4db09c6cec
commit 924fd806d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions

View File

@ -28,7 +28,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#include <stdint.h> #include <stdint.h>
#include "../common.h" #include "../common.h"
#define SGEMM BLASFUNC(sgemm) #define SGEMM BLASFUNC(sgemm)
#define SHGEMM BLASFUNC(shgemm) #define SBGEMM BLASFUNC(sbgemm)
typedef union typedef union
{ {
unsigned short v; unsigned short v;
@ -102,7 +102,7 @@ main (int argc, char *argv[])
} }
SGEMM (&transA, &transB, &m, &n, &k, &alpha, A, SGEMM (&transA, &transB, &m, &n, &k, &alpha, A,
&m, B, &k, &beta, C, &m); &m, B, &k, &beta, C, &m);
SHGEMM (&transA, &transB, &m, &n, &k, &alpha, AA, SBGEMM (&transA, &transB, &m, &n, &k, &alpha, AA,
&m, BB, &k, &beta, CC, &m); &m, BB, &k, &beta, CC, &m);
for (i = 0; i < n; i++) for (i = 0; i < n; i++)
for (j = 0; j < m; j++) for (j = 0; j < m; j++)
@ -126,6 +126,6 @@ main (int argc, char *argv[])
} }
} }
if (ret != 0) if (ret != 0)
fprintf (stderr, "FATAL ERROR SHGEMM - Return code: %d\n", ret); fprintf (stderr, "FATAL ERROR SBGEMM - Return code: %d\n", ret);
return ret; return ret;
} }