forked from mtx512/rk3588-npu
-
Notifications
You must be signed in to change notification settings - Fork 0
/
meson.build
33 lines (28 loc) · 2.38 KB
/
meson.build
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
project('rk3588-npu', 'c')
incdir = include_directories('include')
lib_src = ['src/npu_interface.c','src/npu_matmul.c']
lib = library('rk3588-npu',lib_src, include_directories : incdir)
test_matmul_4_36_16 = executable('matmul_4_36_16', 'tests/matmul_4_36_16.c', include_directories : incdir, link_with : lib)
test('matmul_4x36x16',test_matmul_4_36_16, is_parallel : false)
# Test inputs fp16 and output fp32
test_matmul_fp16 = executable('matmul_fp16', 'tests/matmul_fp16.c', include_directories : incdir, link_with : lib, link_args : '-lm')
test('matmul fp16 1x32x16',test_matmul_fp16, is_parallel : false , args : ['1', '32' ,'16'])
test('matmul fp16 1x64x64',test_matmul_fp16, is_parallel : false , args : ['1', '64' ,'64'])
test('matmul fp16 1x1024x1024',test_matmul_fp16, is_parallel : false , args : ['1', '1024' ,'1024'])
test('matmul fp16 1x4096x4096',test_matmul_fp16, is_parallel : false , args : ['1', '4096' ,'4096'])
test('matmul fp16 4x32x16',test_matmul_fp16, is_parallel : false , args : ['4', '32' ,'16'])
# test max feature data for one task
test('matmul fp16 384x384x4096',test_matmul_fp16, is_parallel : false , args : ['384', '384' ,'4096'])
test_matmul_int8 = executable('matmul_int8', 'tests/matmul_int8.c', include_directories : incdir, link_with : lib, link_args : '-lm')
test('matmul int8 1x32x32',test_matmul_int8, is_parallel : false , args : ['1','32','32'])
test('matmul int8 1x64x64',test_matmul_int8, is_parallel : false , args : ['1','64','64'])
test('matmul int8 1x1024x1024',test_matmul_int8, is_parallel : false , args : ['1','1024','1024'])
test('matmul int8 1x4096x4096',test_matmul_int8, is_parallel : false , args : ['1','4096','4096'])
# test max feature data for one task
test('matmul int8 544x544x4096',test_matmul_int8, is_parallel : false , args : ['544','544','4096'])
# Test inputs fp16 and output fp16
test_matmul_fp16_fp16 = executable('matmul_fp16_fp16', 'tests/matmul_fp16_fp16.c', include_directories : incdir, link_with : lib, link_args : '-lm')
test('matmul fp16_fp16 1x32x16',test_matmul_fp16_fp16, is_parallel : false , args : ['1', '32' ,'16'])
test('matmul fp16_fp16 1x768x768',test_matmul_fp16_fp16, is_parallel : false , args : ['1', '768' ,'768'])
test('matmul fp16_fp16 1x768x2048',test_matmul_fp16_fp16, is_parallel : false , args : ['1', '768' ,'2048'])
test('matmul fp16_fp16 1x8192x8192',test_matmul_fp16_fp16, is_parallel : false , args : ['1', '8192' ,'8192'])