gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[general] added simple jit interface

Browse Source
This commit is contained in:
Philippe Tillet
2019-03-08 23:58:42 -05:00
parent c5073a5af6
commit d049679aa2
26 changed files with 458 additions and 464 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lib/driver/platform.cpp
View File

@@ -31,20 +31,20 @@ namespace triton
namespace driver
{
std::string Platform::version() const{
std::string platform::version() const{
int version;
dispatch::cuDriverGetVersion(&version);
return std::to_string(version);
}
std::vector<Device> Platform::devices() const{
std::vector<Device> devices;
std::vector<device> platform::devices() const{
std::vector<device> devices;
int N;
dispatch::cuDeviceGetCount(&N);
for(int i = 0 ; i < N ; ++i){
CUdevice device;
dispatch::cuDeviceGet(&device, i);
devices.push_back(Device(device));
CUdevice dvc;
dispatch::cuDeviceGet(&dvc, i);
devices.push_back(driver::device(dvc));
}
return devices;
}