#!/bin/bash

set -e

export CUDA_VISIBLE_DEVICES=0

# Smaller batch size so that sequences with Flair embeddings fit in GPU memory.

python -m scripts.benchmark benchmark_gpu \

    --bilstmcrf_large_batch_size 64 \

    --bilstmcrf_fast_batch_size 64

export CUDA_VISIBLE_DEVICES=""

export MKL_NUM_THREADS=32

python -m scripts.benchmark benchmark_cpu_32_threads

export CUDA_VISIBLE_DEVICES=""

export MKL_NUM_THREADS=16

python -m scripts.benchmark benchmark_cpu_16_threads

export CUDA_VISIBLE_DEVICES=""

export MKL_NUM_THREADS=8

python -m scripts.benchmark benchmark_cpu_8_threads