Skip to content

FeixLiu/torch_mlperf_code

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

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

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published