ROCm support using hip - Githubissues

Jittor / jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

https://cg.cs.tsinghua.edu.cn/jittor/

Apache License 2.0

3.08k stars 310 forks source link

ROCm support using hip #129

Open littlewu2508 opened 4 years ago

littlewu2508 commented 4 years ago

Hi! Greetings from Centor fro High Energy Physics, Tsinghua University.

Currently we succesfully depoyed pytorch with the latest rocm-3.5.0 support on Gentoo. Pytorch uses a python script build_amd.py to convert all cuda codes into hip codes (called hipify). So, with rocm developer tools including hipcc and rocm libraries, torch can be compiled and run on AMD GPU without modification of pytorch front-end. So would jittor consider including support on ROCm?

cjld commented 4 years ago

This sounds interesting, we hope Jittor can support AMD GPU, you can add my wechat(randonlang) for further discussing