Skip to content

Navigation Menu

Sign in
Sign up

Latest commit

History

4 Commits

Folders and files

NameName
Last commit message
Last commit date

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

1 star

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages

AltStyle によって変換されたページ (->オリジナル) /