From 198994d01d4e226fd3a8816e8b2fad00c84a691c Mon Sep 17 00:00:00 2001
From: Nhat Nguyen <nhat-nguyen@users.noreply.github.com>
Date: Tue, 4 Apr 2023 20:34:13 -0400
Subject: [PATCH] Register PytorchAtenDomain in RegisterOrtOpSchemas (#14567)

---
 orttraining/orttraining/core/graph/training_op_defs.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/orttraining/orttraining/core/graph/training_op_defs.cc b/orttraining/orttraining/core/graph/training_op_defs.cc
index b2f87213bd..b669b4bb6d 100644
--- a/orttraining/orttraining/core/graph/training_op_defs.cc
+++ b/orttraining/orttraining/core/graph/training_op_defs.cc
@@ -4583,6 +4583,7 @@ void RegisterOrtOpSchemas() {
   }
   domainToVersionRangeInstance.AddDomainToVersion(onnxruntime::kMSExperimentalDomain, 1, 1);
   domainToVersionRangeInstance.AddDomainToVersion(onnxruntime::kMSNchwcDomain, 1, 1);
+  domainToVersionRangeInstance.AddDomainToVersion(onnxruntime::kPytorchAtenDomain, 1, 1);
 
   onnxruntime::contrib::RegisterContribSchemas();
   onnxruntime::training::RegisterTrainingOpSchemas();