zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-28 22:28:11 +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

1
benchmarks/numpy/single_ops.py
View File

@@ -1,7 +1,6 @@
# Copyright © 2023 Apple Inc.
import numpy as np
from time_utils import time_fn

2
benchmarks/python/batch_matmul_bench.py
View File

@@ -1,8 +1,8 @@
# Copyright © 2023 Apple Inc.
import argparse
import mlx.core as mx
import mlx.core as mx
from time_utils import time_fn
B = 8

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"

2
benchmarks/python/llama_torch_bench.py
View File

@@ -4,8 +4,8 @@ import math
import time
import torch
import torch.nn as nn
import torch.mps
import torch.nn as nn
def sync_if_needed(x):

2
benchmarks/python/single_ops.py
View File

@@ -1,8 +1,8 @@
# Copyright © 2023 Apple Inc.
import argparse
import mlx.core as mx
import mlx.core as mx
from time_utils import time_fn