6 lines
95 B
C
6 lines
95 B
C
|
#include <torch/extension.h>
|
||
|
|
|
||
|
|
int get_device_attribute(
|
||
|
|
int attribute,
|
||
|
|
int device_id);
|