From 5be29684b8453b93f115c9ca3113856dd7858f43 Mon Sep 17 00:00:00 2001 From: Yang Wei Date: Wed, 2 Nov 2022 17:28:54 +0800 Subject: [PATCH] Fix gpu name (#415) * fix gpu name bug when on different gpus * update version to 3.0.1 --- lightseq/__init__.py | 2 +- lightseq/csrc/kernels/cuda_util.cu | 2 +- lightseq/inference/tools/util.cc.cu | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/lightseq/__init__.py b/lightseq/__init__.py index 528787cf..05527687 100644 --- a/lightseq/__init__.py +++ b/lightseq/__init__.py @@ -1 +1 @@ -__version__ = "3.0.0" +__version__ = "3.0.1" diff --git a/lightseq/csrc/kernels/cuda_util.cu b/lightseq/csrc/kernels/cuda_util.cu index bdc5cd78..30a89dea 100644 --- a/lightseq/csrc/kernels/cuda_util.cu +++ b/lightseq/csrc/kernels/cuda_util.cu @@ -312,7 +312,7 @@ std::string getGPUName() { return name; } } - throw std::runtime_error("Not support this kind of GPU!"); + return ""; } void print_time_duration( diff --git a/lightseq/inference/tools/util.cc.cu b/lightseq/inference/tools/util.cc.cu index 6202e436..8f337ed2 100644 --- a/lightseq/inference/tools/util.cc.cu +++ b/lightseq/inference/tools/util.cc.cu @@ -366,7 +366,7 @@ std::string getGPUName() { return name; } } - throw std::runtime_error("Not support this kind of GPU!"); + return ""; } } // namespace cuda