Skip to content

tiran/elfgpu

Repository files navigation

ELF GPU

Work-in-progress / early research phase

The goal of this project is to extend pyelftools so it can inspect CUDA and HIP/ROCm fat ELF binaries and detect which GPU devices (architectures) are supported by a given shared library or executable.

See REFERENCES.md for binary format documentation, tools, and research links.

Example output for Torch 2.11.0

torch-2.11.0-cp312-cp312-manylinux_2_28_x86_64.whl

$ python3 -m elfgpu libtorch_cuda.so 
sm_100
sm_100a
sm_103a
sm_120
sm_120a
sm_121a
sm_75
sm_80
sm_86
sm_89
sm_90
sm_90a

torch-2.11.0+rocm7.2-cp312-cp312-manylinux_2_28_x86_64.whl

$ python3 -m elfgpu libtorch_hip.so 
gfx1030
gfx1100
gfx1101
gfx1102
gfx1150
gfx1151
gfx1200
gfx1201
gfx900
gfx906
gfx908
gfx90a
gfx942
gfx950

About

pyelftools extension for CUDA / ROCm ELF fatbins

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages