已合并
delete dropout effect #45755
lihui创建于 9月3日
delete dropout effect #45755
已合并
共 1 个文件变更+0-9
| @@ -630,15 +630,6 @@ def _patch_model_24(): | |||
| 630 | anir_config.force_fallback_kernel_names["mlir_fused_add_lt_neg_where_13"] = True | 630 | anir_config.force_fallback_kernel_names["mlir_fused_add_lt_neg_where_13"] = True |
| 631 | except ImportError: | 631 | except ImportError: |
| 632 | log.warning("import config failed for T5ForConditionalGeneration patch") | 632 | log.warning("import config failed for T5ForConditionalGeneration patch") |
| 633 | - from torch._higher_order_ops.effects import ( | ||
| 634 | - _EffectType, | ||
| 635 | - _register_effectful_op, | ||
| 636 | - ) | ||
| 637 | - | ||
| 638 | - _register_effectful_op( | ||
| 639 | - torch.ops.aten.native_dropout.default, | ||
| 640 | - _EffectType.ORDERED, | ||
| 641 | - ) | ||
| 642 | 633 | ||
| 643 | 634 | ||
| 644 | @register_patch("BartForCausalLM") | 635 | @register_patch("BartForCausalLM") |