Python get_device_count Beispiele

Beispiel #1

Datei: test_reset_device.py Projekt: Alexhuszagh/numba

 def newthread(exception_queue):
     try:
         devices = range(driver.get_device_count())
         for _ in range(2):
             for d in devices:
                 cuda.select_device(d)
                 cuda.close()
     except Exception as e:
         exception_queue.put(e)

Beispiel #2

Datei: test_reset_device.py Projekt: ASPP/numba

 def newthread():
     devices = range(driver.get_device_count())
     print('Devices', devices)
     for _ in range(2):
         for d in devices:
             cuda.select_device(d)
             print('Selected device', d)
             cuda.close()
             print('Closed device', d)

Beispiel #3

Datei: test_reset_device.py Projekt: ChiLi90/LifetimeFit

 def newthread(exception_queue):
     try:
         devices = range(driver.get_device_count())
         for _ in range(2):
             for d in devices:
                 cuda.select_device(d)
                 cuda.close()
     except Exception as e:
         exception_queue.put(e)

Beispiel #4

Datei: test_cuda_driver.py Projekt: winstonewert/numba

 def setUp(self):
     self.assertTrue(driver.get_device_count())
     device = driver.get_device()
     ccmajor, _ = device.compute_capability
     if ccmajor >= 2:
         self.ptx = ptx2
     else:
         self.ptx = ptx1
     self.context = device.get_or_create_context()

Beispiel #5

Datei: test_cuda_driver.py Projekt: ASPP/numba

 def setUp(self):
     self.assertTrue(driver.get_device_count())
     device = driver.get_device()
     ccmajor, _ = device.compute_capability
     if ccmajor >= 2:
         self.ptx = ptx2
     else:
         self.ptx = ptx1
     self.context = device.get_or_create_context()

Beispiel #6

Datei: test_reset_device.py Projekt: winstonewert/numba

 def newthread():
     devices = range(driver.get_device_count())
     print('Devices', devices)
     for _ in range(2):
         for d in devices:
             cuda.select_device(d)
             print('Selected device', d)
             cuda.close()
             print('Closed device', d)

Beispiel #7

 def newthread(exception_queue):
     try:
         devices = range(driver.get_device_count())
         print('Devices', devices)
         for _ in range(2):
             for d in devices:
                 cuda.select_device(d)
                 print('Selected device', d)
                 cuda.close()
                 print('Closed device', d)
     except Exception as e:
         exception_queue.put(e)

Beispiel #8

Datei: test_reset_device.py Projekt: ymarfoq/outilACVDesagregation

 def newthread(exception_queue):
     try:
         devices = range(driver.get_device_count())
         print('Devices', devices)
         for _ in range(2):
             for d in devices:
                 cuda.select_device(d)
                 print('Selected device', d)
                 cuda.close()
                 print('Closed device', d)
     except Exception as e:
         exception_queue.put(e)