FilesExpand file tree

 master

/

device_query.cpp

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

23 lines (18 loc) · 457 Bytes

 master

/

device_query.cpp

Copy path

Top

File metadata and controls

• Code
• Blame

23 lines (18 loc) · 457 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

// Copyright 2014 BVLC and contributors.

#include "caffe/common.hpp"

#include "caffe/net.hpp"

using namespace caffe; // NOLINT(build/namespaces)

int main(int argc, char** argv) {

if (argc > 2) {

LOG(ERROR) << "device_query [device_id=0]";

return 1;

}

if (argc == 2) {

LOG(INFO) << "Querying device_id=" << argv[1];

Caffe::SetDevice(atoi(argv[1]));

Caffe::DeviceQuery();

} else {

Caffe::DeviceQuery();

}

return 0;

}