[Fix][Unity][Training] Fix a bug in Gradient tests due to merging

mlc-ai / relax

Apache License 2.0

140 stars 70 forks source link

[Fix][Unity][Training] Fix a bug in Gradient tests due to merging #241

Closed Ubospica closed 1 year ago

Ubospica commented 1 year ago

There was a conflict in the modification about the Gradient on the mlc and unity branches, which resulted in testing errors. This PR fixes the issue in tests.

Ubospica commented 1 year ago

cc @tqchen