Rename "HALF" and "sh" to "BFLOAT16" and "sb"
This commit is contained in:
parent
4db09c6cec
commit
924fd806d0
|
@ -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;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue