#37616 [ROCm][CI] Fix flaky Cohere/OpenAI embedding parity test
原始 PR · 作者 AndreasKaratzas · 合并时间 2026-03-25 18:55
为 ROCm 平台添加 ROCM_EXTRA_ARGS 到两个测试文件并早期验证 pooling params,修复 flaky 的嵌入测试。
建议关注测试设计的权衡,尤其是 `ROCM_EXTRA_ARGS` 的使用是否应更明确标注或跳过相关测试。同时,异常处理更改值得了解,可参考相关 issue 以跟踪批不变性问题进展。
参与讨论