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: