csullivan/note.md

Last active September 7, 2023 06:50

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/csullivan/a7cbe5ed233b9a6a5c634537c0c3e905.js"></script>
Save csullivan/a7cbe5ed233b9a6a5c634537c0c3e905 to your computer and use it in GitHub Desktop.

Download ZIP

CUTLASS CMake configuration for Hopper (sm90a)

Raw

note.md

Reference: https://github.com/NVIDIA/cutlass/blob/e01b9b5029b7caca5a43c29f7d2714d7cf1dcae8/README.md?plain=1#L137C10-L137C37

cmake -DCUTLASS_LIBRARY_KERNELS=cutlass3x_sm90* -DCUTLASS_ENABLE_CUBLAS=ON -DCUTLASS_ENABLE_CUDNN=ON -DCUTLASS_NVCC_ARCHS="90a" ..

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment