From 22841d772d5125ea2b9e34808bc0298d7b7fd5f2 Mon Sep 17 00:00:00 2001 From: drisspg Date: Sun, 19 Oct 2025 17:58:31 +0000 Subject: [PATCH] Update (base update) [ghstack-poisoned] --- torch/_inductor/codegen/triton.py | 1 + torch/_inductor/runtime/triton_heuristics.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/torch/_inductor/codegen/triton.py b/torch/_inductor/codegen/triton.py index 856aadbe93ee..aa230918188d 100644 --- a/torch/_inductor/codegen/triton.py +++ b/torch/_inductor/codegen/triton.py @@ -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: diff --git a/torch/_inductor/runtime/triton_heuristics.py b/torch/_inductor/runtime/triton_heuristics.py index 709f0ec8b11a..2dc48b1b45f8 100644 --- a/torch/_inductor/runtime/triton_heuristics.py +++ b/torch/_inductor/runtime/triton_heuristics.py @@ -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: