Skip to content

Commit 5be2968

Browse files
authored
Fix gpu name (#415)
* fix gpu name bug when on different gpus * update version to 3.0.1
1 parent 07ff5a2 commit 5be2968

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

lightseq/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "3.0.0"
1+
__version__ = "3.0.1"

lightseq/csrc/kernels/cuda_util.cu

+1-1
Original file line numberDiff line numberDiff line change
@@ -312,7 +312,7 @@ std::string getGPUName() {
312312
return name;
313313
}
314314
}
315-
throw std::runtime_error("Not support this kind of GPU!");
315+
return "";
316316
}
317317

318318
void print_time_duration(

lightseq/inference/tools/util.cc.cu

+1-1
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,7 @@ std::string getGPUName() {
366366
return name;
367367
}
368368
}
369-
throw std::runtime_error("Not support this kind of GPU!");
369+
return "";
370370
}
371371

372372
} // namespace cuda

0 commit comments

Comments
 (0)