add test_sqrt to test_speed_v_torch (tinygrad#9066)

working on getting rid of llvm sqrt hack
chenyuxyz · Feb 13, 2025 · 947c97e · 947c97e
1 parent 49abc09
commit 947c97e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/test_speed_v_torch.py b/test/test_speed_v_torch.py
@@ -190,6 +190,10 @@ def test_exp(self):
     def f(a, b): return a.exp()
     helper_test_generic_square('exp', 2048, f, f, onearg=True)
 
+  def test_sqrt(self):
+    def f(a, b): return a.sqrt()
+    helper_test_generic_square('sqrt', 2048, f, f, onearg=True)
+
   def test_relu(self):
     def f(a, b): return a.relu()
     helper_test_generic_square('relu', 4096, f, f, onearg=True)