forked from ccf-ai-infra/GPUCodeForces
10 lines
334 B
Markdown
10 lines
334 B
Markdown
|
# 文件说明
|
|||
|
|
|
|||
|
|
example_torchcode.py:torch代码示例
|
|||
|
|
|
|||
|
|
example_cudacode.py：和torch对应的cuda代码
|
|||
|
|
|
|||
|
|
prompt.txt：利用LLM从torch代码生成cuda代码的prompt示例，（原始torch代码被附在prompt最后）
|
|||
|
|
|
|||
|
|
run_code.py：用于测试生成的cuda代码和原始torch输出是否一致以及加速情况的示例代码
|