update scale dot assertion in plot_layout.py#774

jtang10 wants to merge 1 commit intomain_perfROCm/triton:main_perffrom

jingtang/update_scaleDot_plotLayoutROCm/triton:jingtang/update_scaleDot_plotLayoutCopy head branch name to clipboard