zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 04:24:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add isort pre-commit and run (#68)

Browse Source
This commit is contained in:
Zach Schillaci
2023-12-08 20:31:47 +01:00
committed by GitHub
parent e89c571de7
commit 5b9be57ac3
33 changed files with 54 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
benchmarks/python/blas/bench_gemm.py
View File

@@ -1,13 +1,14 @@
# Copyright © 2023 Apple Inc.
import numpy as np
import argparse
import mlx.core as mx
import time
import torch
import os
import math
import os
import subprocess
import time
import mlx.core as mx
import numpy as np
import torch
device_name = subprocess.check_output(["sysctl", "-n", "machdep.cpu.brand_string"])
device_name = device_name.decode("utf-8").strip("\n")

10
benchmarks/python/blas/bench_gemv.py
View File

@@ -1,14 +1,14 @@
# Copyright © 2023 Apple Inc.
import matplotlib.pyplot as plt
import numpy as np
import argparse
import mlx.core as mx
import time
import torch
import os
import subprocess
import time
import matplotlib.pyplot as plt
import mlx.core as mx
import numpy as np
import torch
results_dir = "./results"