InfiniTensor/include/utils
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
..
broadcast_shape.h add infer index function (#175) 2023-11-24 09:24:25 +08:00
data_convert.h support mixed dtype (#102) 2023-08-16 21:49:43 +08:00
data_generator.h 支持fp16 dtype (#96) 2023-08-02 16:38:16 +08:00
dataloader.h Json perfrecord (#32) 2022-09-22 15:34:34 +08:00
exception.h tensor parallel for transformer (#125) 2023-09-14 14:19:45 +08:00
operator_utils.h Add layer normalization (#181) 2023-11-24 15:15:14 +08:00
small_array.h tensor parallel for transformer (#125) 2023-09-14 14:19:45 +08:00
validation.h Verify wanghailu 0902 (#22) 2022-09-05 15:45:52 +08:00