|
|
||
|---|---|---|
| .. | ||
| example_cudacode.py | ||
| example_torchcode.py | ||
| prompt.txt | ||
| readme.md | ||
| run_code.py | ||
| 参赛者需要提供的内容.md | ||
readme.md
文件说明
example_torchcode.py:torch代码示例
example_cudacode.py:和torch对应的cuda代码
prompt.txt:利用LLM从torch代码生成cuda代码的prompt示例,(原始torch代码被附在prompt最后)
run_code.py:用于测试生成的cuda代码和原始torch输出是否一致以及加速情况的示例代码