Skip to content

Commit 854b31d

Browse files
author
lsf
committed
lsf pc update
1 parent e6831f3 commit 854b31d

File tree

9 files changed

+13
-464
lines changed

9 files changed

+13
-464
lines changed

main.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,6 @@ void test_track(){
4242

4343
int main(){
4444
test_yolo();
45-
test_track();
45+
// test_track();
4646
return 0;
4747
}

trt_common/cuda_tools.cpp

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -50,18 +50,6 @@ namespace CUDATools{
5050
return numJobs < GPU_BLOCK_THREADS ? numJobs : GPU_BLOCK_THREADS;
5151
}
5252

53-
std::string device_capability(int device_id){
54-
cudaDeviceProp prop{};
55-
checkCudaRuntime(cudaGetDeviceProperties(&prop, device_id));
56-
return iLogger::format("%d.%d", prop.major, prop.minor);
57-
}
58-
59-
std::string device_name(int device_id){
60-
cudaDeviceProp prop{};
61-
checkCudaRuntime(cudaGetDeviceProperties(&prop, device_id));
62-
return prop.name;
63-
}
64-
6553
std::string device_description(){
6654

6755
cudaDeviceProp prop{};

trt_common/cuda_tools.hpp

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -54,14 +54,11 @@ namespace CUDATools{
5454
dim3 grid_dims(int numJobs);
5555
dim3 block_dims(int numJobs);
5656

57-
// return 8.6 etc.
58-
std::string device_capability(int device_id);
59-
std::string device_name(int device_id);
6057
std::string device_description();
6158

6259
class AutoDevice{
6360
public:
64-
AutoDevice(int device_id = 0);
61+
explicit AutoDevice(int device_id = 0);
6562
virtual ~AutoDevice();
6663

6764
private:

0 commit comments

Comments
 (0)