Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/jatinx/pyhip
Python Interface to HIP and hiprtc Library
https://github.com/jatinx/pyhip
bindings cuda gpu hip hiprtc python rocm
Last synced: 3 months ago
JSON representation
Python Interface to HIP and hiprtc Library
- Host: GitHub
- URL: https://github.com/jatinx/pyhip
- Owner: jatinx
- License: mit
- Created: 2021-10-31T09:06:51.000Z (over 3 years ago)
- Default Branch: main
- Last Pushed: 2023-11-19T23:27:10.000Z (about 1 year ago)
- Last Synced: 2024-03-25T13:06:06.802Z (10 months ago)
- Topics: bindings, cuda, gpu, hip, hiprtc, python, rocm
- Language: Python
- Homepage:
- Size: 65.4 KB
- Stars: 6
- Watchers: 1
- Forks: 5
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# PyHIP - Python Interface of [HIP](https://github.com/ROCm-Developer-Tools/HIP) and [hip-on-nv](https://github.com/jatinx/hip-on-nv)
## Introduction
This small python lib which hooks into HIP library and provides pythonic interface to it.
There are two parts of it, hip library and hiprtc library.
This also works with [hip-on-nv](https://github.com/jatinx/hip-on-nv).
## Installation
### Prerequisites
Before installing pyhip-interface, please make sure you have the following prerequisites installed:
* The Heterogeneous Interface for Portability - [HIP Installation Guide](https://docs.amd.com/bundle/HIP-Installation-Guide-v5.3/page/Introduction_to_HIP_Installation_Guide.html)
### Install with pip
PyHIP can be installed using pip, a package manager for Python. Run the following command in your terminal to install:
`pip install pyhip-interface`
### Install from source
Alternatively, you can install PyHIP from the source code. First, clone the repository from GitHub:
`git clone https://github.com/jatinx/PyHIP`
Then, navigate to the repository directory and run the following command to install:
`python setup.py install`
## Example Usage
```python
import ctypes
from pyhip import hip, hiprtc
source = """
extern "C" __global__ void set(int *a) {
*a = 10;
}
"""
prog = hiprtc.hiprtcCreateProgram(source, 'set', [], [])
device_properties = hip.hipGetDeviceProperties(0)
print(f"Compiling kernel for {device_properties.gcnArchName}")
if hip.hipGetPlatformName() == 'amd': # offload arch is amd only
hiprtc.hiprtcCompileProgram(prog, [f'--offload-arch={device_properties.gcnArchName}'])
else:
hiprtc.hiprtcCompileProgram(prog, [])
code = hiprtc.hiprtcGetCode(prog)
module = hip.hipModuleLoadData(code)
kernel = hip.hipModuleGetFunction(module, 'set')
ptr = hip.hipMalloc(4)
class PackageStruct(ctypes.Structure):
_fields_ = [("a", ctypes.c_void_p)]
struct = PackageStruct(ptr)
hip.hipModuleLaunchKernel(kernel, 1, 1, 1, 1, 1, 1, 0, 0, struct)
res = ctypes.c_int(0)
hip.hipMemcpy_dtoh(ctypes.byref(res), ptr, 4)
print(res.value)
```
## Testing
Testing is in it's infancy. I'll try to add more as and when I get some time. Any help here is appreciated :)
### How to run tests
Make sure you have ```pytest``` package installed.
From the project folder run ```pytest ./tests/* -v -r A```
## Help needed
If you are an opensource developer and want to contribute to this project, I can use your help in several places depending on your skillset.
- Adding more tests to the project
- Adding API documentation and samples
- Improving HIP API coverage - adding meaningful APIs instead of direct mapping.
## Common Problems
- Unable to load hip or hiprtc library - OSError: libamdhip64.so: cannot open shared object file: No such file or directory.
- Make sure that LD_LIBRARY_PATH has hip path on it (/opt/rocm/lib or a custom installation path)
- Getting error and need to debug
- Set AMD_LOG_LEVEL=7 for maximum verbosity of HIP APIs (this only works on amd platform).