mirror of
https://github.com/pmeier/light-the-torch.git
synced 2024-09-08 23:29:28 +03:00
73e8fc80fe
* Support multiple CUDA versions * cleanup * exclude specific windows combinations from tests * fix cpu backend ordering * prioritize backend over version * fix docstring
18 lines
332 B
Python
18 lines
332 B
Python
import pytest
|
|
|
|
import light_the_torch.computation_backend as cb
|
|
|
|
|
|
class GenericComputationBackend(cb.ComputationBackend):
|
|
@property
|
|
def local_specifier(self):
|
|
return "generic"
|
|
|
|
def __lt__(self, other):
|
|
return NotImplemented
|
|
|
|
|
|
@pytest.fixture
|
|
def generic_backend():
|
|
return GenericComputationBackend()
|