Merge nucleic/sleek-ember-seal-uady into dev
This commit is contained in:
@@ -9,6 +9,29 @@ sys.path.insert(0, str(MODULE_DIR))
|
||||
import eval as purpose_eval
|
||||
|
||||
|
||||
class CoreMLComputeUnitTests(unittest.TestCase):
|
||||
class CoreMLTools:
|
||||
class ComputeUnit:
|
||||
ALL = "all-value"
|
||||
CPU_ONLY = "cpu-value"
|
||||
CPU_AND_GPU = "gpu-value"
|
||||
CPU_AND_NE = "ne-value"
|
||||
|
||||
def test_maps_cli_compute_policies(self):
|
||||
expected = {
|
||||
"all": "all-value",
|
||||
"cpu-only": "cpu-value",
|
||||
"cpu-and-gpu": "gpu-value",
|
||||
"cpu-and-ne": "ne-value",
|
||||
}
|
||||
for requested, value in expected.items():
|
||||
with self.subTest(requested=requested):
|
||||
self.assertEqual(
|
||||
value,
|
||||
purpose_eval._coreml_compute_unit(self.CoreMLTools, requested),
|
||||
)
|
||||
|
||||
|
||||
class TierDriftTests(unittest.TestCase):
|
||||
def test_current_routing_matrix_bounds_every_label_pair(self):
|
||||
records = [{"prompt": f"prompt {index}"} for index in range(8 * 8)]
|
||||
|
||||
Reference in New Issue
Block a user