generate_dataset.sh 1.2 KB

12345678910111213141516171819202122232425262728
  1. #!/bin/bash
  2. export LOGOPS=/tmp/ops
  3. rm $LOGOPS
  4. # generate many kernels
  5. PYTHONPATH="." OPT=2 GPU=1 python3 test/external/external_test_opt.py
  6. PYTHONPATH="." OPT=3 GPU=1 python3 test/external/external_test_opt.py
  7. GPU=1 IMAGE=1 python3 test/test_ops.py
  8. FORWARD_ONLY=1 GPU=1 IMAGE=2 python test/test_ops.py
  9. STEPS=3 python3 examples/hlb_cifar10.py
  10. WINO=1 STEPS=3 python3 examples/hlb_cifar10.py
  11. python3 examples/stable_diffusion.py --noshow
  12. python3 examples/llama.py --prompt "hello" --count 5
  13. python3 examples/gpt2.py --count 5
  14. HALF=1 python3 examples/gpt2.py --count 5
  15. python3 examples/beautiful_mnist.py
  16. python3 examples/beautiful_cartpole.py
  17. python3 examples/mlperf/model_spec.py
  18. python3 examples/yolov8.py ./test/models/efficientnet/Chicken.jpg
  19. examples/openpilot/go.sh
  20. JIT=2 BIG=1 MPS=1 pytest test/ --ignore=test/test_fusion_op.py --ignore=test/test_linearizer_failures.py --ignore=test/test_gc.py --ignore=test/test_speed_v_torch.py --ignore=test/test_jit.py
  21. JIT=2 BIG=1 MPS=1 python -m pytest test/test_gc.py
  22. JIT=2 BIG=1 MPS=1 python -m pytest test/test_jit.py
  23. JIT=2 BIG=1 MPS=1 python -m pytest test/test_speed_v_torch.py
  24. # sort and uniq
  25. sort -u /tmp/ops > /tmp/sops
  26. ls -lh /tmp/ops /tmp/sops