InfiniTensor/test/operators
xgqdut2016 a7293c12ba
Add layer normalization (#181)
* - add layernorm kernel

* success:add layernorm kernel and test

* fix: remove unusalble comments

* fix: modify code as reviewer suggested

* debug,modified .cu and test

* optional bias support

* overloading function

* fix bug after merging; remove time constrain in conv test

---------

Co-authored-by: kilinchange <kilinchange@163.com>
Co-authored-by: Haojie Wang <haojie0429@gmail.com>
2023-11-24 15:15:14 +08:00
..
test_all_gather.cc impl distributed launch with NCCL (#106) 2023-09-05 09:47:35 +08:00
test_all_reduce.cc impl distributed launch with NCCL (#106) 2023-09-05 09:47:35 +08:00
test_batch_norm.cc refactor(core): 添加新的 `OpType` 定义 (#99) 2023-08-07 11:17:05 +08:00
test_broadcast.cc impl distributed launch with NCCL (#106) 2023-09-05 09:47:35 +08:00
test_clip.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_concat.cc support Dynamic tensor infer shape and fix memory pool (#176) 2023-11-23 13:11:50 +08:00
test_conv.cc Add layer normalization (#181) 2023-11-24 15:15:14 +08:00
test_conv_transposed_2d.cc ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00
test_element_wise.cc ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00
test_expand.cc 框架支持bert/gpt2模型构图 (#94) 2023-08-29 16:06:52 +08:00
test_extend.cc ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00
test_gather.cc 框架支持bert/gpt2模型构图 (#94) 2023-08-29 16:06:52 +08:00
test_gather_elements.cc Add GatherElements op and cuda kernel (#149) 2023-10-12 09:18:12 +08:00
test_matmul.cc support mixed dtype (#102) 2023-08-16 21:49:43 +08:00
test_pad.cc ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00
test_pooling.cc Pooling ceil mode (#155) 2023-10-09 20:51:39 +08:00
test_reduce.cc Add ReduceSum op and kernel (#160) 2023-11-24 09:29:58 +08:00
test_reshape.cc Cpu backend2 (#77) 2023-04-17 12:15:23 +08:00
test_resize.cc ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00
test_slice.cc Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
test_split.cc impl sqrt on CUDA (#109) 2023-08-18 12:17:47 +08:00
test_transpose.cc support mixed dtype (#102) 2023-08-16 21:49:43 +08:00
test_unary.cc 【Hackathon No.108】Add Gelu operator, ffi, kernel for cpu and gpu. (#148) 2023-10-10 15:21:13 +08:00
test_where.cc 框架支持bert/gpt2模型构图 (#94) 2023-08-29 16:06:52 +08:00