benchmark_attnmask.py 14.2 KB