GPUCodeForces/example/001-example/readme.md

10 lines
334 B
Markdown
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

# 文件说明
example_torchcode.py:torch代码示例
example_cudacode.py和torch对应的cuda代码
prompt.txt利用LLM从torch代码生成cuda代码的prompt示例原始torch代码被附在prompt最后
run_code.py用于测试生成的cuda代码和原始torch输出是否一致以及加速情况的示例代码