Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
cuda_utils.h 109 Bytes
Newer Older
Mingcan Xiang's avatar
Avoid multiple redefinition (#1817)  
Mingcan Xiang committed
1
2 
#pragma once
Yanming W's avatar
[Build] Avoid building too many extensions (#1624)  
Yanming W committed
3
4
5
6
7 
#include <torch/extension.h>

int get_device_attribute(
    int attribute,
    int device_id);