From 3c262d5ef81be6e273a21e7ce5a2803476c27dfd Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Fri, 1 Nov 2024 11:52:01 -0700 Subject: [PATCH] Reapply "Change compute capablity min value (#7708)" This reverts commit 69e85e2893dbaf5d144a13ea43a53426a7558a4c. --- CMakeLists.txt | 4 ++-- qa/L0_java_memory_growth/MemoryGrowthTest.java | 4 ++-- qa/L0_java_resnet/ResnetTest.java | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 56cb346dc0..e4cc8ef481 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -1,4 +1,4 @@ -# Copyright 2020-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# Copyright 2020-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions @@ -42,7 +42,7 @@ option(TRITON_ENABLE_NVTX "Include NVTX support in server" OFF) option(TRITON_ENABLE_GPU "Enable GPU support in server" ON) option(TRITON_ENABLE_MALI_GPU "Enable Arm Mali GPU support in server" OFF) option(TRITON_IGPU_BUILD "Enable options for iGPU compilation in sever" OFF) -set(TRITON_MIN_COMPUTE_CAPABILITY "6.0" CACHE STRING +set(TRITON_MIN_COMPUTE_CAPABILITY "7.5" CACHE STRING "The minimum CUDA compute capability supported by Triton" ) set(TRITON_EXTRA_LIB_PATHS "" CACHE PATH "Extra library paths for Triton Server build") diff --git a/qa/L0_java_memory_growth/MemoryGrowthTest.java b/qa/L0_java_memory_growth/MemoryGrowthTest.java index 28243459ec..522bba584f 100644 --- a/qa/L0_java_memory_growth/MemoryGrowthTest.java +++ b/qa/L0_java_memory_growth/MemoryGrowthTest.java @@ -1,4 +1,4 @@ -// Copyright 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +// Copyright 2022-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved. // // Redistribution and use in source and binary forms, with or without // modification, are permitted provided that the following conditions @@ -34,7 +34,7 @@ import org.bytedeco.tritonserver.tritonserver.*; public class MemoryGrowthTest { - static final double TRITON_MIN_COMPUTE_CAPABILITY = 6.0; + static final double TRITON_MIN_COMPUTE_CAPABILITY = 7.5; private static boolean done = false; static float max_growth_allowed = .10f; static int max_mem_allowed = 30; diff --git a/qa/L0_java_resnet/ResnetTest.java b/qa/L0_java_resnet/ResnetTest.java index 4827273926..37a8906930 100644 --- a/qa/L0_java_resnet/ResnetTest.java +++ b/qa/L0_java_resnet/ResnetTest.java @@ -1,4 +1,4 @@ -// Copyright 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +// Copyright 2022-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved. // // Redistribution and use in source and binary forms, with or without // modification, are permitted provided that the following conditions @@ -41,7 +41,7 @@ public class ResnetTest { // TODO: fix build to support GPU only resnet50v1.5_fp16_savedmodel //"resnet50v1.5_fp16_savedmodel", }; - private static final double TRITON_MIN_COMPUTE_CAPABILITY = 6.0; + private static final double TRITON_MIN_COMPUTE_CAPABILITY = 7.5; private enum Backend { NONE, ONNX,