History

Kuohais 9bf06d7e4b 强调提交规范		2025-11-13 16:19:48 +08:00
..
example_cudacode.py	强调提交规范	2025-11-13 16:19:48 +08:00
example_torchcode.py	强调提交规范	2025-11-13 16:19:48 +08:00
prompt.txt	强调提交规范	2025-11-13 16:19:48 +08:00
readme.md	强调提交规范	2025-11-13 16:19:48 +08:00
run_code.py	强调提交规范	2025-11-13 16:19:48 +08:00
参赛者需要提供的内容.md	强调提交规范	2025-11-13 16:19:48 +08:00

readme.md

文件说明

example_torchcode.py:torch代码示例

example_cudacode.py：和torch对应的cuda代码

prompt.txt：利用LLM从torch代码生成cuda代码的prompt示例，（原始torch代码被附在prompt最后）

run_code.py：用于测试生成的cuda代码和原始torch输出是否一致以及加速情况的示例代码