From d5943704860e9b7baf92e4d16b1d0644abf36b8d Mon Sep 17 00:00:00 2001
From: Richard Barnes <rbarnes@meta.com>
Date: Mon, 13 May 2024 17:58:51 -0700
Subject: [PATCH] c10::optional -> std::optional in
 aiplatform/distributed_inference/pytorch/ops/tests/distributed_inference_operators_test_fixture.h
 +20 (#1007)

Summary:
Pull Request resolved: https://github.com/facebookincubator/AITemplate/pull/1007

`c10::optional` was switched to be `std::optional` after PyTorch moved to C++17. Let's eliminate `c10::optional`, if we can.

Reviewed By: albanD, swolchok

Differential Revision: D57294299

fbshipit-source-id: 7637845c1195ec8a321a833942505122ff53baee
---
 fx2ait/fx2ait/csrc/AITModel.cpp     |  4 ++--
 fx2ait/fx2ait/csrc/AITModel.h       |  4 ++--
 fx2ait/fx2ait/csrc/AITModelImpl.cpp |  4 ++--
 fx2ait/fx2ait/csrc/AITModelImpl.h   | 12 ++++++------
 4 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/fx2ait/fx2ait/csrc/AITModel.cpp b/fx2ait/fx2ait/csrc/AITModel.cpp
index eed7855c4..510bbb534 100644
--- a/fx2ait/fx2ait/csrc/AITModel.cpp
+++ b/fx2ait/fx2ait/csrc/AITModel.cpp
@@ -62,8 +62,8 @@ static auto registerAITModel =
              std::string,
              std::vector<std::string>,
              std::vector<std::string>,
-             c10::optional<at::ScalarType>,
-             c10::optional<at::ScalarType>,
+             std::optional<at::ScalarType>,
+             std::optional<at::ScalarType>,
              int64_t>())
         .def("forward", &AITModel::forward)
         .def("profile", &AITModel::profile)
diff --git a/fx2ait/fx2ait/csrc/AITModel.h b/fx2ait/fx2ait/csrc/AITModel.h
index 4780b7ed1..01efa1300 100644
--- a/fx2ait/fx2ait/csrc/AITModel.h
+++ b/fx2ait/fx2ait/csrc/AITModel.h
@@ -25,8 +25,8 @@ class AITModel : public torch::CustomClassHolder {
       const std::string& model_path,
       std::vector<std::string> input_names,
       std::vector<std::string> output_names,
-      c10::optional<at::ScalarType> input_dtype,
-      c10::optional<at::ScalarType> output_dtype,
+      std::optional<at::ScalarType> input_dtype,
+      std::optional<at::ScalarType> output_dtype,
       int64_t num_runtimes = 2,
       bool use_cuda_graph = false)
       : aitModelImpl_(
diff --git a/fx2ait/fx2ait/csrc/AITModelImpl.cpp b/fx2ait/fx2ait/csrc/AITModelImpl.cpp
index 739e1935e..74586cb6e 100644
--- a/fx2ait/fx2ait/csrc/AITModelImpl.cpp
+++ b/fx2ait/fx2ait/csrc/AITModelImpl.cpp
@@ -133,8 +133,8 @@ AITModelImpl::AITModelImpl(
     const std::string& model_path,
     std::vector<std::string> input_names,
     std::vector<std::string> output_names,
-    c10::optional<at::ScalarType> input_dtype,
-    c10::optional<at::ScalarType> output_dtype,
+    std::optional<at::ScalarType> input_dtype,
+    std::optional<at::ScalarType> output_dtype,
     int64_t num_runtimes,
     bool use_cuda_graph)
     : handle_(dlopen(model_path.c_str(), RTLD_NOW | RTLD_LOCAL)),
diff --git a/fx2ait/fx2ait/csrc/AITModelImpl.h b/fx2ait/fx2ait/csrc/AITModelImpl.h
index 063808b83..196c39ae9 100644
--- a/fx2ait/fx2ait/csrc/AITModelImpl.h
+++ b/fx2ait/fx2ait/csrc/AITModelImpl.h
@@ -43,8 +43,8 @@ class AITModelImpl {
       const std::string& model_path,
       std::vector<std::string> input_names,
       std::vector<std::string> output_names,
-      c10::optional<at::ScalarType> input_dtype,
-      c10::optional<at::ScalarType> output_dtype,
+      std::optional<at::ScalarType> input_dtype,
+      std::optional<at::ScalarType> output_dtype,
       int64_t num_runtimes = 2,
       bool use_cuda_graph = false);
 
@@ -102,11 +102,11 @@ class AITModelImpl {
     return output_names_;
   }
 
-  const c10::optional<at::ScalarType> floatingPointInputDtype() const {
+  const std::optional<at::ScalarType> floatingPointInputDtype() const {
     return floating_point_input_dtype_;
   }
 
-  const c10::optional<at::ScalarType> floatingPointOutputDtype() const {
+  const std::optional<at::ScalarType> floatingPointOutputDtype() const {
     return floating_point_output_dtype_;
   }
 
@@ -182,8 +182,8 @@ class AITModelImpl {
   const std::string library_path_;
   const std::vector<std::string> input_names_;
   const std::vector<std::string> output_names_;
-  const c10::optional<at::ScalarType> floating_point_input_dtype_;
-  const c10::optional<at::ScalarType> floating_point_output_dtype_;
+  const std::optional<at::ScalarType> floating_point_input_dtype_;
+  const std::optional<at::ScalarType> floating_point_output_dtype_;
 #ifdef FBCODE_AIT
   folly::F14FastMap<const char*, size_t> input_name_to_index_;
   folly::F14FastMap<const char*, size_t> output_name_to_index_;