frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update (base update)

Browse Source 
[ghstack-poisoned]
This commit is contained in:
drisspg
2025-10-19 17:58:31 +00:00
parent 36371b8ec7
commit 22841d772d
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/_inductor/codegen/triton.py
View File

@ -4729,6 +4729,7 @@ class TritonKernel(SIMDKernel[TritonCSEVariable]):
"spill_threshold": config.triton.spill_threshold,
"store_cubin": config.triton.store_cubin,
"deterministic": config.deterministic,
"force_filter_reduction_configs": config.test_configs.force_filter_reduction_configs,
}
if config.write_are_deterministic_algorithms_enabled:

2
torch/_inductor/runtime/triton_heuristics.py
View File

@ -2973,7 +2973,7 @@ def filter_reduction_configs_for_determinism(
def _do_filter_due_to_inductor_config():
return (
inductor_meta.get("deterministic", False)
or torch._inductor.config.test_configs.force_filter_reduction_configs
or inductor_meta.get("force_filter_reduction_configs", False)
) or inductor_meta.get("are_deterministic_algorithms_enabled")
if not _do_filter_due_to_inductor_config() or len(configs) == 1: