kailums 1b38c05544 change ci docker image to rocm6.1 (#21296) ... ### Description <!-- Describe your changes. --> There is a bug for kernel running on rocm6.0, so change ci docker image to rocm6.1 For the torch installed in the docker image, change to rocm repo when it is not 6.0 version. ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->		2024-07-18 14:50:01 +08:00
..
github	change ci docker image to rocm6.1 (#21296)	2024-07-18 14:50:01 +08:00
__init__.py
amd_hipify.py	[ROCm] Add SkipGroupNorm for ROCm EP (#19303)	2024-02-21 11:08:48 +08:00
build.py	Migraphx ep windows build (#21284)	2024-07-11 21:21:38 -07:00
clean_docker_image_cache.py	Bump ruff to 0.3.2 and black to 24 (#19878)	2024-03-13 10:00:32 -07:00
compile_triton.py
coverage.py
gen_def.py	Exclude azure ep from gen_def.cc (#21250)	2024-07-04 10:50:27 -07:00
get_docker_image.py	Bump ruff to 0.3.2 and black to 24 (#19878)	2024-03-13 10:00:32 -07:00
logger.py
op_registration_utils.py	Bump ruff to 0.3.2 and black to 24 (#19878)	2024-03-13 10:00:32 -07:00
op_registration_validator.py	Bump ruff to 0.3.2 and black to 24 (#19878)	2024-03-13 10:00:32 -07:00
patch_manylinux.py
policheck_exclusions.xml
reduce_op_kernels.py
replace_urls_in_deps.py
requirements-transformers-test.txt	test: refactor flash_attn tests to use parameterized (#20913)	2024-06-11 15:57:20 -07:00
set-trigger-rules.py	Add VP test in Stable diffusion pipeline (#19300)	2024-01-29 09:33:58 -08:00
update_tsaoptions.py
upload_python_package_to_azure_storage.py