Skip to content

Commit f3d84ff

Browse files
committed
Add UUID to environment for cupy and nvcuda backends
1 parent 32c3d8a commit f3d84ff

2 files changed

Lines changed: 7 additions & 0 deletions

File tree

kernel_tuner/backends/cupy.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from __future__ import print_function
33

44
import numpy as np
5+
import uuid
56

67
from kernel_tuner.backends.backend import GPUBackend
78
from kernel_tuner.observers.cupy import CupyRuntimeObserver
@@ -80,6 +81,10 @@ def __init__(self, device=0, iterations=7, compiler_options=None, observers=None
8081
env["iterations"] = self.iterations
8182
env["compiler_options"] = compiler_options
8283
env["device_properties"] = self.devprops
84+
85+
props = cp.cuda.runtime.getDeviceProperties(device)
86+
env["uuid"] = str(uuid.UUID(bytes=props["uuid"]))
87+
8388
self.env = env
8489
self.name = env["device_name"]
8590

kernel_tuner/backends/nvcuda.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from warnings import warn
33

44
import numpy as np
5+
import uuid
56

67
from kernel_tuner.backends.backend import GPUBackend
78
from kernel_tuner.observers.nvcuda import CudaRuntimeObserver
@@ -99,6 +100,7 @@ def __init__(self, device=0, iterations=7, compiler_options=None, observers=None
99100
err, device_properties = runtime.cudaGetDeviceProperties(device)
100101
cuda_error_check(err)
101102
env = dict()
103+
env["uuid"] = str(uuid.UUID(bytes=device_properties.uuid.bytes))
102104
env["device_name"] = device_properties.name.decode()
103105
env["pci_bus_id"] = device_properties.pciBusID
104106
env["cuda_version"] = driver.CUDA_VERSION

0 commit comments

Comments
 (0)