Add softmin, hardshrink, hardtanh (#1180)

--------- Co-authored-by: Nikhil Mehta <[email protected]>
2024-06-04 15:48:18 -07:00
parent 83b11bc58d
commit 0b7d71fd2f
14 changed files with 110 additions and 20 deletions
@@ -35,7 +35,6 @@ def run_bench(system_size):


 def time_fft():
-
    with mx.stream(mx.cpu):
        cpu_bandwidths = run_bench(system_size=int(2**22))