From 3ae2bfaefe4344820d0a17055aa23f080530af34 Mon Sep 17 00:00:00 2001 From: Abhishek Jindal Date: Thu, 10 Mar 2022 11:56:30 -0800 Subject: [PATCH] Abjindal/torch api change gelu (#10833) * changing gelu backward op and adding required files * cleaning up file and adding comments * version comparison issue --- orttraining/orttraining/eager/opgen/opgen/atenops.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/orttraining/orttraining/eager/opgen/opgen/atenops.py b/orttraining/orttraining/eager/opgen/opgen/atenops.py index d2d3076455..1d80f2a48b 100644 --- a/orttraining/orttraining/eager/opgen/opgen/atenops.py +++ b/orttraining/orttraining/eager/opgen/opgen/atenops.py @@ -11,7 +11,7 @@ from opgen.onnxops import * import torch from packaging import version -TORCH_API_CHANGE_VERSION = "1.11.0" +TORCH_API_CHANGE_VERSION = "1.11.1" kMSDomain = 'onnxruntime::kMSDomain' @@ -101,7 +101,7 @@ hand_implemented = { # Signature of gelu_backward was changed in this commit id 983ba5e585485ed61a0c0012ef6944f5685e3d97 and PR 61439 # This is done to make sure it is backward and future compatible -if version.parse(torch.__version__) <= version.parse(TORCH_API_CHANGE_VERSION): +if version.parse(torch.__version__) < version.parse(TORCH_API_CHANGE_VERSION): hand_implemented['aten::gelu_backward'] = GeluGrad('grad', 'self') else: hand_implemented['aten::gelu_backward'] = GeluGrad('grad_output', 'self')