Skip to content

[NVIDIA] Add Jetson Orin/Thor/Spark/GB300 Codegen #7337

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 9 commits into from
Closed

[NVIDIA] Add Jetson Orin/Thor/Spark/GB300 Codegen #7337

Changes from 7 commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
49cfb70
Update CMakeLists.txt
johnnynunez Jun 19, 2025
00a9411
Update CMakeLists.txt
johnnynunez Jun 19, 2025
fb0a89e
move inside aarch64
johnnynunez Jun 19, 2025
b245a44
Merge branch 'main' into patch-4
johnnynunez Jun 23, 2025
8a1cea5
Merge branch 'main' into patch-4
johnnynunez Aug 20, 2025
4d65464
Merge branch 'main' into patch-4
johnnynunez Aug 20, 2025
42bed8a
Merge branch 'main' into patch-4
FlamingoPg Aug 23, 2025
ab4d232
Merge branch 'main' into patch-4
johnnynunez Aug 23, 2025
2e891e1
Merge branch 'main' into patch-4
FlamingoPg Aug 25, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 12 additions & 2 deletions sgl-kernel/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -171,16 +171,25 @@ option(SGL_KERNEL_ENABLE_FA3 "Enable FA3" OFF)
if (ENABLE_BELOW_SM90)
list(APPEND SGL_KERNEL_CUDA_FLAGS
"-gencode=arch=compute_80,code=sm_80"
"-gencode=arch=compute_87,code=sm_87"
"-gencode=arch=compute_89,code=sm_89"
)
endif()

if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "13.0" OR SGL_KERNEL_ENABLE_SM100A)
list(APPEND SGL_KERNEL_CUDA_FLAGS
"-gencode=arch=compute_103,code=sm_103"
"-gencode=arch=compute_103a,code=sm_103a"
"-gencode=arch=compute_110,code=sm_110"
"-gencode=arch=compute_110a,code=sm_110a"
"-gencode=arch=compute_121,code=sm_121"
"-gencode=arch=compute_121a,code=sm_121a"
)
endif()
if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8" OR SGL_KERNEL_ENABLE_SM100A)
list(APPEND SGL_KERNEL_CUDA_FLAGS
"-gencode=arch=compute_100,code=sm_100"
"-gencode=arch=compute_100a,code=sm_100a"
"-gencode=arch=compute_101,code=sm_101"
"-gencode=arch=compute_101a,code=sm_101a"
"-gencode=arch=compute_120,code=sm_120"
"-gencode=arch=compute_120a,code=sm_120a"
)
Expand Down Expand Up @@ -374,6 +383,7 @@ if (SGL_KERNEL_ENABLE_FA3)
list(APPEND SGL_FLASH_KERNEL_CUDA_FLAGS
"-gencode=arch=compute_80,code=sm_80"
"-gencode=arch=compute_86,code=sm_86"
"-gencode=arch=compute_87,code=sm_87"
)
# SM8X Logic
file(GLOB FA3_SM8X_GEN_SRCS
Expand Down
Loading