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	[Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789)	2023-07-21 12:53:41 -07:00
coverage.py	Format all python files under onnxruntime with black and isort (#11324)	2022-04-26 09:35:16 -07:00
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	Format all python files under onnxruntime with black and isort (#11324)	2022-04-26 09:35:16 -07:00
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	[Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789)	2023-07-21 12:53:41 -07:00
policheck_exclusions.xml	Exculde hipify option from policheck (#13431)	2022-10-25 16:35:16 +08:00
reduce_op_kernels.py	Re-organize the transpose optimization and layout transformation files. (#16246)	2023-07-07 08:24:47 +10:00
replace_urls_in_deps.py	Add a build validation for Linux ARM64 cross-compile (#18200)	2023-11-08 13:03:18 -08:00
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	Adopt linrtunner as the linting tool - take 2 (#15085)	2023-03-24 15:29:03 -07:00
upload_python_package_to_azure_storage.py	[Linter] Bump ruff and remove pylint (#17797)	2023-10-05 21:07:33 -07:00