Skip to content

Instantly share code, notes, and snippets.

@yaroslavvb2

yaroslavvb2/matmul_bench.sh

Last active November 3, 2017 02:16
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save yaroslavvb2/b275194b4808d32abc8cc1c4398b531b to your computer and use it in GitHub Desktop.
Save yaroslavvb2/b275194b4808d32abc8cc1c4398b531b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
matmul_bench.sh
wget https://raw.githubusercontent.com/yaroslavvb/stuff/master/matmul_benchmark_seq.py
export TF_CPP_MIN_LOG_LEVEL=1
export CUDA_VISIBLE_DEVICES=0
python matmul_benchmark_seq.py --dtype=float16
python matmul_benchmark_seq.py --dtype=float32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment