apache--tvm

github-featured/apache--tvm

Fork 0

Commit Graph

Author	SHA1	Message	Date
Tianqi Chen	aa2e609136	[LINT] Modernize lint to use pre-commit hooks (#18807 ) This PR migrates existing lint to use pre-commit hooks	2026-02-22 11:03:21 -05:00
Bohan Hou	f67657fe09	[Unity][TOPI] fp16 LayerNorm & GroupNorm (#14264 ) This pr modifies the topi implementation (which is also the legalizer's backend of Relax) of LayerNorm and GroupNorm operators to allow them to accept fp16 inputs, cast to fp32 internally, and produce fp16 outputs. This can help eliminate unnecessary casts caused by AMP.	2023-04-01 15:31:44 -04:00
Wuwei Lin	4e783a6087	[TOPI] Add layer norm operator (#12864 ) * [TOPI] Add one-pass layer norm using tuple reduction * Add reducer pattern for LowerCrossThreadReduction * lint * update docs	2022-09-22 13:20:40 -07:00

Author

SHA1

Message

Date

Tianqi Chen

aa2e609136

[LINT] Modernize lint to use pre-commit hooks (#18807 )

This PR migrates existing lint to use pre-commit hooks

2026-02-22 11:03:21 -05:00

Bohan Hou

f67657fe09

[Unity][TOPI] fp16 LayerNorm & GroupNorm (#14264 )

This pr modifies the topi implementation (which is also the legalizer's backend of Relax) of LayerNorm and GroupNorm operators to allow them to accept fp16 inputs, cast to fp32 internally, and produce fp16 outputs.

This can help eliminate unnecessary casts caused by AMP.

2023-04-01 15:31:44 -04:00

Wuwei Lin

4e783a6087

[TOPI] Add layer norm operator (#12864 )

* [TOPI] Add one-pass layer norm using tuple reduction

* Add reducer pattern for LowerCrossThreadReduction

* lint

* update docs

2022-09-22 13:20:40 -07:00

3 Commits