Fix gpu name (#415)

* fix gpu name bug when on different gpus * update version to 3.0.1
bytedance · Nov 2, 2022 · 5be2968 · 5be2968
1 parent 07ff5a2
commit 5be2968
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -366,7 +366,7 @@ std::string getGPUName() {
       return name;
     }
   }
-  throw std::runtime_error("Not support this kind of GPU!");
+  return "";
 }
 
 }  // namespace cuda