ccf-ai-infra
/
GPUCodeForces
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 Commits
1 Branch
2.6 MB
1 Download
Tree: 3dcc7faeaa
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '3dcc7faeaa'
${ noResults }
GPUCodeForces/example/001-example/readme.md

readme.md 334 B
Raw Normal View History

example of GPUCodeForces
3 months ago
​
​
​
​
​
​
​
​
 123456789 
  1. # 文件说明

  2. 

  3. example_torchcode.py:torch代码示例

  4. 

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

  6. 

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

  8. 

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

No Description