cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

mma.c (964B)

      1// SPDX-License-Identifier: GPL-2.0-or-later
      2/*
      3 * Test basic matrix multiply assist (MMA) functionality if available.
      4 *
      5 * Copyright 2020, Alistair Popple, IBM Corp.
      6 */
      7#include <stdio.h>
      8#include <stdint.h>
      9
     10#include "utils.h"
     11
     12extern void test_mma(uint16_t (*)[8], uint16_t (*)[8], uint32_t (*)[4*4]);
     13
     14static int mma(void)
     15{
     16	int i;
     17	int rc = 0;
     18	uint16_t x[] = {1, 0, 2, 0, 3, 0, 4, 0};
     19	uint16_t y[] = {1, 0, 2, 0, 3, 0, 4, 0};
     20	uint32_t z[4*4];
     21	uint32_t exp[4*4] = {1, 2, 3, 4,
     22			     2, 4, 6, 8,
     23			     3, 6, 9, 12,
     24			     4, 8, 12, 16};
     25
     26	SKIP_IF_MSG(!have_hwcap2(PPC_FEATURE2_ARCH_3_1), "Need ISAv3.1");
     27	SKIP_IF_MSG(!have_hwcap2(PPC_FEATURE2_MMA), "Need MMA");
     28
     29	test_mma(&x, &y, &z);
     30
     31	for (i = 0; i < 16; i++) {
     32		printf("MMA[%d] = %d ", i, z[i]);
     33
     34		if (z[i] == exp[i]) {
     35			printf(" (Correct)\n");
     36		} else {
     37			printf(" (Incorrect)\n");
     38			rc = 1;
     39		}
     40	}
     41
     42	return rc;
     43}
     44
     45int main(int argc, char *argv[])
     46{
     47	return test_harness(mma, "mma");
     48}