[Tests] Fix tests 1st ed. - Githubissues

FlagOpen / FlagGems

FlagGems is an operator library for large language models implemented in Triton Language.

Apache License 2.0

347 stars 48 forks source link

[Tests] Fix tests 1st ed. #208

Closed zhzhcookie closed 2 months ago

zhzhcookie commented 2 months ago

PR Category

[OP Test]

Type of Change

[Tests]

Description

Modify test cases, reduce similar type test cases, increase test cases of different dimensions and size ranges, and try to unify the test shapes of similar operators.
Fix --device=cpu.

Issue

Progress

[ ] Change is properly reviewed (1 reviewer required, 2 recommended).
[ ] Change is responded to an issue.
[ ] Change is fully covered by a UT.

Performance