Skip to content

Conversation

@pengcheng888
Copy link
Contributor

目标版本
main

功能描述
添加python的scaled_dot_product_attention的实现和测试

测试结果
sdpa_cpu
Screenshot from 2025-11-13 08-42-42
sdpa_nvidia
Screenshot from 2025-11-13 08-42-39

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[DEV] 添加python的scaled_dot_product_attention的实现和测试

1 participant