export ROCBLAS_LAYER=3


python trition_opt.py