Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
cuda_utils.h 95 Bytes
Newer Older
Yanming W's avatar
[Build] Avoid building too many extensions (#1624)  
Yanming W committed
1
2
3
4
5 
#include <torch/extension.h>

int get_device_attribute(
    int attribute,
    int device_id);