mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-02-12 15:45:27 -05:00
a66ffec6d3
disk tensor load contains big offset and is not meant to be run by gpu. repro steps ``` time ./extra/optimization/generate_dataset.sh gzip /tmp/sops mv /tmp/sops.gz extra/datasets/ ```
26 lines
858 B
Bash
Executable File
26 lines
858 B
Bash
Executable File
#!/bin/bash
|
|
export LOGOPS=/tmp/ops
|
|
rm $LOGOPS
|
|
|
|
# generate many kernels
|
|
PYTHONPATH="." OPT=2 GPU=1 python3 test/external/external_test_opt.py
|
|
PYTHONPATH="." OPT=3 GPU=1 python3 test/external/external_test_opt.py
|
|
GPU=1 IMAGE=1 python3 test/test_ops.py
|
|
FORWARD_ONLY=1 GPU=1 IMAGE=2 python test/test_ops.py
|
|
STEPS=3 python3 examples/hlb_cifar10.py
|
|
WINO=1 STEPS=3 python3 examples/hlb_cifar10.py
|
|
python3 examples/stable_diffusion.py --noshow
|
|
python3 examples/llama.py --prompt "hello" --count 5
|
|
python3 examples/gpt2.py --count 5
|
|
HALF=1 python3 examples/gpt2.py --count 5
|
|
python3 examples/beautiful_mnist.py
|
|
python3 examples/beautiful_cartpole.py
|
|
python3 examples/mlperf/model_spec.py
|
|
python3 examples/yolov8.py ./test/models/efficientnet/Chicken.jpg
|
|
openpilot/go.sh
|
|
BIG=1 MPS=1 pytest test/
|
|
|
|
# sort and uniq
|
|
sort -u /tmp/ops > /tmp/sops
|
|
ls -lh /tmp/ops /tmp/sops
|