diff --git a/CMakeLists.txt b/CMakeLists.txt
index 3bd7d1ee..5464eda9 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -1141,6 +1141,7 @@ if(MGE_WITH_DISTRIBUTED)
     set(MEGRAY_WITH_NCCL ${MGE_WITH_CUDA} CACHE BOOL "Override MegRay option" FORCE)
     set(MEGRAY_WITH_SHM ${MGE_WITH_CUDA} CACHE BOOL "Override MegRay option" FORCE)
     set(MEGRAY_WITH_RCCL ${MGE_WITH_ROCM} CACHE BOOL "Override MegRay option" FORCE)
+    set(MEGRAY_CUDA_GENCODE ${MGE_CUDA_GENCODE} CACHE STRING "Overwrite MegRay CUDA -gencode specifications" FORCE)
     add_subdirectory(${PROJECT_SOURCE_DIR}/third_party/MegRay)
 endif()