1  /* { dg-do run { target { ! ia32 } } } */
       2  /* { dg-require-effective-target amx_complex } */
       3  /* { dg-require-effective-target avx512fp16 } */
       4  /* { dg-options "-O2 -mamx-complex -mavx512fp16" } */
       5  #define AMX_COMPLEX
       6  #define DO_TEST test_amx_complex_cmmrlfp16ps
       7  void test_amx_complex_cmmrlfp16ps();
       8  #include "amx-helper.h"
       9  
      10  void calc_matrix_cmmrlfp16ps (__tile *dst, __tile *src1, __tile *src2)
      11  {
      12    uint16_t *src1_buf = (uint16_t *) src1->buf;
      13    uint16_t *src2_buf = (uint16_t *) src2->buf;
      14    float *dst_buf = (float *) dst->buf;
      15    
      16    int M = src1->rows;
      17    int N = src1->colsb / 4;
      18    int K = src2->colsb / 4;
      19    int i, j, k, t;
      20  
      21    for (i = 0; i < M; i++)
      22      for (j = 0; j < N; j++)
      23        for (k = 0; k < K; k++)
      24  	for (t = 0; t < 2; t+=2)
      25  	  dst_buf[i * N + k] += 
      26  	    (make_fp16_f32 (src1_buf[i * 2 * N + 2 * j + t]) *
      27  	      make_fp16_f32 (src2_buf[j * 2 * K + 2 * k + t])) -
      28  	    (make_fp16_f32 (src1_buf[i * 2 * N + 2 * j + t + 1]) *
      29  	      make_fp16_f32 (src2_buf[j * 2 * K + 2 * k + t + 1]));
      30  }
      31  
      32  void test_amx_complex_cmmrlfp16ps ()
      33  {
      34    __tilecfg_u cfg;
      35    __tile dst, dst_ref, src1, src2;
      36    uint8_t tmp_dst_buf[1024], tmp_dst_zero_buf[1024];
      37  
      38    init_fp16_max_tile_buffer (tmp_dst_buf);
      39    init_fp16_max_tile_zero_buffer (tmp_dst_zero_buf);
      40  
      41    init_tile_config (&cfg);
      42    init_tile_reg_and_src_with_buffer (1, dst, tmp_dst_zero_buf);
      43    init_tile_reg_and_src_with_buffer (2, src1, tmp_dst_buf);
      44    init_tile_reg_and_src_with_buffer (3, src2, tmp_dst_buf);
      45  
      46    calc_matrix_cmmrlfp16ps (&dst, &src1, &src2);
      47    
      48    _tile_cmmrlfp16ps (1, 2, 3);
      49    _tile_stored (1, dst_ref.buf, _STRIDE);
      50  
      51    if (!check_tile_register (&dst_ref, &dst))
      52          abort ();
      53  }