Skip to content

Latest commit

 

History

History
22 lines (18 loc) · 585 Bytes

README.md

File metadata and controls

22 lines (18 loc) · 585 Bytes
Raw

code from mlperf v1.0 result

cd 1_node_8_A100_PyTorch

use these commands to compile mha relating op

cd ./mhalib
python setup.py build
cp build/lib*/mhalib* ../
cd ..

run with this command

bash run_and_time.sh

note

  1. the data root path should be /workspace/bert_data
  2. only update the extra params in config_DGXA100_1x8x56x1.sh, for other config file, follow the same extra_params in config_DGXA100_1x8x56x1.sh