From fc03fc025e14c404c8bc5dc8ded53ba2899d5896 Mon Sep 17 00:00:00 2001 From: chenyu Date: Fri, 5 Jul 2024 14:52:09 -0400 Subject: [PATCH] enable sin on METAL in test_dtype_alu (#5298) --- test/test_dtype_alu.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/test/test_dtype_alu.py b/test/test_dtype_alu.py index 23ad91146e..5039b962f3 100644 --- a/test/test_dtype_alu.py +++ b/test/test_dtype_alu.py @@ -40,8 +40,7 @@ unary_operations = [(Tensor.exp, np.exp), (Tensor.log, np.log), operator.neg, (T #binary_operations += [(Tensor.maximum, np.maximum)] # TODO: CUDACPU segfaults on sin -# TODO: METAL sin is flaky for float16 -if getenv("CUDACPU") or (getenv("MOCKGPU") and Device.DEFAULT == "NV") or Device.DEFAULT == "METAL": unary_operations.remove((Tensor.sin, np.sin)) +if getenv("CUDACPU") or (getenv("MOCKGPU") and Device.DEFAULT == "NV"): unary_operations.remove((Tensor.sin, np.sin)) class ht: float64 = strat.floats(width=64, allow_subnormal=False)