File: cpu_multi_head_attention.cpp

package info (click to toggle)
miopen 6.4.3%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 66,788 kB
  • sloc: cpp: 300,511; lisp: 29,731; ansic: 2,683; sh: 471; python: 323; makefile: 155
file content (17 lines) | stat: -rw-r--r-- 445 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
 
#include "cpu_multi_head_attention.hpp"

struct CPU_Mha_FP32 : test::cpu::CPUMHATest<float, float>
{
};

struct CPU_Mha_FP8 : test::cpu::CPUMHATest<float, float8>
{
};

TEST_P(CPU_Mha_FP32, CPUMHATestFloatFw) {}

TEST_P(CPU_Mha_FP8, CPUMHATestFloat8Fw) {}

INSTANTIATE_TEST_SUITE_P(Smoke, CPU_Mha_FP32, testing::ValuesIn(test::cpu::CPUMHAConfigs()));

INSTANTIATE_TEST_SUITE_P(Smoke, CPU_Mha_FP8, testing::ValuesIn(test::cpu::CPUMHAConfigs()));