mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-04 05:10:06 +06:00
706703bba6
* Add expectation classes + tests * Use typing Union instead of | * Use bits to track score in properties cmp method * Add exceptions and tests + comments * Remove compute cap minor as it is not needed currently * Simplify. Remove Properties class * Add example Exceptions usage * Expectations as dict subclass * Update example Exceptions usage * Refactor. Improve type name. Document score fn. * Rename to DeviceProperties.
33 lines
921 B
Python
33 lines
921 B
Python
import unittest
|
|
|
|
from transformers.testing_utils import Expectations
|
|
|
|
|
|
class ExpectationsTest(unittest.TestCase):
|
|
def test_expectations(self):
|
|
expectations = Expectations(
|
|
{
|
|
(None, None): 1,
|
|
("cuda", 8): 2,
|
|
("cuda", 7): 3,
|
|
("rocm", 8): 4,
|
|
("rocm", None): 5,
|
|
("cpu", None): 6,
|
|
}
|
|
)
|
|
|
|
def check(value, key):
|
|
assert expectations.find_expectation(key) == value
|
|
|
|
# xpu has no matches so should find default expectation
|
|
check(1, ("xpu", None))
|
|
check(2, ("cuda", 8))
|
|
check(3, ("cuda", 7))
|
|
check(4, ("rocm", 9))
|
|
check(4, ("rocm", None))
|
|
check(2, ("cuda", 2))
|
|
|
|
expectations = Expectations({("cuda", 8): 1})
|
|
with self.assertRaises(ValueError):
|
|
expectations.find_expectation(("xpu", None))
|