add rapids triton stuff back

cmake/thirdparty/get_rapids-triton.cmake

@@ -40,14 +40,12 @@ endfunction()
 set (RAPIDS_FORK https://github.com/rapidsai/rapids-triton.git)
 set (REPO_TAG branch-${RAPIDS_DEPENDENCIES_VERSION})
 message(STATUS "Setting repo tag to ${REPO_TAG} for rapids fork ${RAPIDS_FORK}")
-# if Triton tag and organization is specified, change the fork and the repo
+# if Triton tag and organization is non-default, change the fork and repo tag used
+# for rapids
 if (NOT RAPIDS_TRITON_REPO_PATH STREQUAL RAPIDS_FORK)
   set (RAPIDS_FORK ${RAPIDS_TRITON_REPO_PATH})
-  message(STATUS "Setting repo fork to ${RAPIDS_FORK}")
-endif()
-if (NOT RAPIDS_TRITON_REPO_TAG STREQUAL "main")
   set (REPO_TAG ${RAPIDS_TRITON_REPO_TAG})
-  message(STATUS "Setting repo tag to ${REPO_TAG}")
+  message(STATUS "Re-setting repo tag to ${REPO_TAG} for rapids fork ${RAPIDS_FORK}")
 endif()
 
 find_and_configure_rapids_triton(VERSION    ${RAPIDS_DEPENDENCIES_VERSION}

ops/Dockerfile

@@ -163,6 +163,8 @@ RUN source /conda/dev/bin/activate \
       -DTRITON_CORE_REPO_TAG="${TRITON_CORE_REPO_TAG}" \
       -DTRITON_COMMON_REPO_TAG="${TRITON_COMMON_REPO_TAG}" \
       -DTRITON_BACKEND_REPO_TAG="${TRITON_BACKEND_REPO_TAG}" \
+      -DRAPIDS_TRITON_REPO_TAG="${RAPIDS_TRITON_REPO_TAG}" \
+      -DRAPIDS_TRITON_REPO_PATH="${RAPIDS_TRITON_REPO_PATH}" \
       -DTRITON_ENABLE_GPU="${TRITON_ENABLE_GPU}" \
       -DTRITON_ENABLE_STATS="${TRITON_ENABLE_STATS}" \
       -DRAPIDS_DEPENDENCIES_VERSION="${RAPIDS_DEPENDENCIES_VERSION}" \