This repository was archived by the owner on May 13, 2026. It is now read-only.

[kernel] added half2 specialization for layernorm kernel#139

Open

dongxianzhe wants to merge 5 commits into

vectorch-ai:mainfrom

dongxianzhe:op/layernorm_kernel

Commits on Apr 27, 2024

[op] optimize layernorm kernel for half2 type
Xianzhe Dong
committed
[ut] add layernorm kernel unitest
Xianzhe Dong
committed
use gtest library rewrite layernorm kernel unitest
Xianzhe Dong
committed
added layernorm kernel half2 unit test using gtest library
Xianzhe Dong
committed

Commits on Apr 28, 2024

[refactor] use torch::tensor to allocate memory in layernorm kernel unitest and just test llm::kernel::layer_norm
Xianzhe Dong
committed