From 9ff7d808eae617ec66107542b709bb68ca9f62ce Mon Sep 17 00:00:00 2001 From: Misha Chornyi <99709299+mc-nv@users.noreply.github.com> Date: Mon, 21 Oct 2024 09:46:49 -0700 Subject: [PATCH] Revert "Change compute capablity min value (#7708)" (#7721) --- 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 e4cc8ef481..56cb346dc0 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -1,4 +1,4 @@ -# Copyright 2020-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# Copyright 2020-2023, 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 "7.5" CACHE STRING +set(TRITON_MIN_COMPUTE_CAPABILITY "6.0" 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 522bba584f..28243459ec 100644 --- a/qa/L0_java_memory_growth/MemoryGrowthTest.java +++ b/qa/L0_java_memory_growth/MemoryGrowthTest.java @@ -1,4 +1,4 @@ -// Copyright 2022-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +// Copyright 2022-2023, 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 = 7.5; + static final double TRITON_MIN_COMPUTE_CAPABILITY = 6.0; 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 37a8906930..4827273926 100644 --- a/qa/L0_java_resnet/ResnetTest.java +++ b/qa/L0_java_resnet/ResnetTest.java @@ -1,4 +1,4 @@ -// Copyright 2022-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +// Copyright 2022-2023, 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 = 7.5; + private static final double TRITON_MIN_COMPUTE_CAPABILITY = 6.0; private enum Backend { NONE, ONNX,