Pass use_triton
flag to LayerNorm module
#336
Merged
use_triton
flag to LayerNorm module
#336