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

[Bugfix] Handle negative input dimension in 'cat' operator #261

Closed 2niuhe closed 1 month ago

2niuhe commented 1 month ago

PR Category

Operator

Type of Change

Bug Fix

Description

This commit converts negative input dimension to positive in the 'cat' operator and adds unit tests to verify this functionality.

Issue

Progress

Performance