From 544c9d9dbae034aa71533ed54218241545f9c304 Mon Sep 17 00:00:00 2001 From: Morgan Funtowicz Date: Thu, 10 Oct 2024 12:36:32 +0000 Subject: [PATCH] (fix): HOPPER_SM_MAJOR is 9 not 8 --- backends/trtllm/include/hardware.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/trtllm/include/hardware.h b/backends/trtllm/include/hardware.h index da0bf4f3..584dd974 100644 --- a/backends/trtllm/include/hardware.h +++ b/backends/trtllm/include/hardware.h @@ -14,7 +14,7 @@ namespace huggingface::hardware::cuda { #define AMPERE_SM_MAJOR 8 -#define HOPPER_SM_MAJOR 8 +#define HOPPER_SM_MAJOR 9 /** * Store information about the version of the CUDA Compute Capabilities detected on the device