Khanin Udomchoksakul
Add optimized CUDA kernels (RMSNorm, MLP) and patched Gemma4 modeling files
42e38d8 verified | from setuptools import setup | |
| setup( | |
| name="mlp_ext", | |
| packages=["mlp_cuda"], | |
| package_data={"mlp_cuda": ["*.so"]}, | |
| ) | |
| from setuptools import setup | |
| setup( | |
| name="mlp_ext", | |
| packages=["mlp_cuda"], | |
| package_data={"mlp_cuda": ["*.so"]}, | |
| ) | |