Loading repository data…
hlky / repository
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
HAIT continues development from facebookincubator/AITemplate.
Major changes include:
Install CUDA Toolkit
For Windows, install Build Tools for Visual Studio 2022 or Visual Studio 2022 (Community, Professional or Enterprise).
If Build Tools or Visual Studio is installed after CUDA Toolkit, re-run CUDA Toolkit installation to get Visual Studio integration. Build Tools only may need CUDA X.Y.props copying manually, refer to CUDA documentation for the default .props install directory and adjust for your Build Tools install directory.
git clone --recursive https://github.com/hlky/AITemplate
cd AITemplate/python
pip install -e .
python setup.py bdist_wheel
pip uninstall -y aitemplate
pip install dist/*.whl
Generally:
With thanks to the original developers and other Meta engineers.
AITemplate is licensed under the Apache 2.0 License.
Selected from shared topics, language and repository description—not editorial ratings.
facebookincubator /
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
Saik0s /
AITemplater is a Python project that leverages AI to code documentation and templates. It streamlines development by automating repetitive tasks.
wubukeneng /
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.