InfiniOps is a high-performance, cross-platform operator library supporting multiple backends: CPU, Nvidia, MetaX, Iluvatar, Hygon, Moore, Cambricon, and more.
- C++17 compatible compiler
- CMake 3.18+
- Python 3.10+
- Hardware-specific SDKs (e.g. CUDA Toolkit, MUSA Toolkit)
Install with pip (recommended):
pip install .This auto-detects available platforms on supported backends. To specify platforms explicitly:
pip install . -C cmake.define.WITH_CPU=ON -C cmake.define.WITH_NVIDIA=ON| Option | Description | Default |
|---|---|---|
-DWITH_CPU=[ON|OFF] |
Compile the CPU implementation | OFF |
-DWITH_NVIDIA=[ON|OFF] |
Compile the Nvidia implementation | OFF |
-DWITH_METAX=[ON|OFF] |
Compile the MetaX implementation | OFF |
-DWITH_ILUVATAR=[ON|OFF] |
Compile the Iluvatar implementation | OFF |
-DWITH_HYGON=[ON|OFF] |
Compile the Hygon implementation | OFF |
-DWITH_MOORE=[ON|OFF] |
Compile the Moore implementation | OFF |
-DWITH_CAMBRICON=[ON|OFF] |
Compile the Cambricon implementation | OFF |
-DWITH_ASCEND=[ON|OFF] |
Compile the Ascend implementation | OFF |
-DAUTO_DETECT_DEVICES=[ON|OFF] |
Auto-detect available platforms | ON |
If no accelerator options are provided and auto-detection finds nothing, WITH_CPU is enabled by default.
For Hygon builds, set DTK_ROOT to the DTK installation root if it is not installed at /opt/dtk. You can override the default DCU arch with -DHYGON_ARCH=<arch> when configuring CMake.
See CONTRIBUTING.md for code style, commit conventions, PR workflow, development guide, and troubleshooting.
See LICENSE.