Rename Google Tensor float truncation type enum and update flag strings.

LiteRT-PiperOrigin-RevId: 820099975

Author: ai-edge-bot

litert/c/options/BUILD

@@ -107,6 +107,7 @@ cc_test(
     srcs = ["litert_google_tensor_options_test.cc"],
     deps = [
         ":litert_google_tensor_options",
+        ":litert_google_tensor_options_type",
         "//litert/c:litert_common",
         "//litert/c:litert_opaque_options",
         "//litert/cc/options:litert_google_tensor_options",

litert/c/options/litert_google_tensor_options_test.cc

@@ -19,6 +19,7 @@
 #include <gtest/gtest.h>
 #include "litert/c/litert_common.h"
 #include "litert/c/litert_opaque_options.h"
+#include "litert/c/options/litert_google_tensor_options_type.h"
 #include "litert/cc/options/litert_google_tensor_options.h"
 #include "litert/test/matchers.h"
 
@@ -108,7 +109,7 @@ TEST(LiteRtGoogleTensorOptionsTest, DumpOpTimings) {
   LiteRtGoogleTensorOptionsTruncationType truncation_type;
   LITERT_ASSERT_OK(LiteRtGoogleTensorOptionsGetFloatTruncationType(
       options_data, &truncation_type));
-  ASSERT_EQ(truncation_type, kLiteRtGoogleTensorFloatTruncationTypeUnspecified);
+  ASSERT_EQ(truncation_type, kLiteRtGoogleTensorFloatTruncationTypeAuto);
 
   LITERT_ASSERT_OK(LiteRtGoogleTensorOptionsSetFloatTruncationType(
       options_data, kLiteRtGoogleTensorFloatTruncationTypeBfloat16));
@@ -128,7 +129,7 @@ TEST(GoogleTensorOptionsTest, CppApi) {
   EXPECT_TRUE(options->GetInt64ToInt32Truncation());
 
   EXPECT_EQ(options->GetFloatTruncationType(),
-            kLiteRtGoogleTensorFloatTruncationTypeUnspecified);
+            kLiteRtGoogleTensorFloatTruncationTypeAuto);
   options->SetFloatTruncationType(
       kLiteRtGoogleTensorFloatTruncationTypeBfloat16);
   EXPECT_EQ(options->GetFloatTruncationType(),

litert/c/options/litert_google_tensor_options_type.h

@@ -18,7 +18,7 @@
 // float_truncation_type -------------------------------------------------------
 
 typedef enum LiteRtGoogleTensorOptionsTruncationType {
-  kLiteRtGoogleTensorFloatTruncationTypeUnspecified = 0,
+  kLiteRtGoogleTensorFloatTruncationTypeAuto = 0,
   kLiteRtGoogleTensorFloatTruncationTypeNoTruncation = 1,
   kLiteRtGoogleTensorFloatTruncationTypeBfloat16 = 2,
   kLiteRtGoogleTensorFloatTruncationTypeHalf = 3,

litert/runtime/litert_google_tensor.h

@@ -22,7 +22,7 @@
 
 struct LiteRtGoogleTensorOptionsT {
   LiteRtGoogleTensorOptionsTruncationType float_truncation_type =
-      kLiteRtGoogleTensorFloatTruncationTypeUnspecified;
+      kLiteRtGoogleTensorFloatTruncationTypeAuto;
   bool int64_to_int32_truncation = false;
   std::string output_dir = "";
   bool dump_op_timings = false;

litert/tools/flags/vendors/BUILD

@@ -87,6 +87,7 @@ cc_library(
         "-DINCLUDE_GOOGLE_TENSOR_RUNTIME_FLAGS",
     ],
     deps = [
+        "//litert/c/options:litert_google_tensor_options_type",
         "//litert/cc:litert_expected",
         "//litert/cc:litert_macros",
         "//litert/cc/options:litert_google_tensor_options",
@@ -124,6 +125,7 @@ cc_test(
     deps = [
         ":google_tensor_flags",
         "//litert/c/options:litert_google_tensor_options",
+        "//litert/c/options:litert_google_tensor_options_type",
         "@com_google_googletest//:gtest_main",
     ],
 )

litert/tools/flags/vendors/google_tensor_flags.cc

@@ -18,6 +18,7 @@
 
 #include "absl/flags/flag.h"  // from @com_google_absl
 #include "absl/strings/string_view.h"  // from @com_google_absl
+#include "litert/c/options/litert_google_tensor_options_type.h"
 #include "litert/cc/litert_expected.h"
 #include "litert/cc/litert_macros.h"
 #include "litert/cc/options/litert_google_tensor_options.h"
@@ -29,15 +30,15 @@
 bool AbslParseFlag(absl::string_view text,
                    LiteRtGoogleTensorOptionsTruncationType* options,
                    std::string* error) {
-  if (text == "unspecified") {
-    *options = kLiteRtGoogleTensorFloatTruncationTypeUnspecified;
+  if (text == "auto") {
+    *options = kLiteRtGoogleTensorFloatTruncationTypeAuto;
     return true;
   }
   if (text == "no_truncation") {
     *options = kLiteRtGoogleTensorFloatTruncationTypeNoTruncation;
     return true;
   }
-  if (text == "bf16") {
+  if (text == "bfloat16") {
     *options = kLiteRtGoogleTensorFloatTruncationTypeBfloat16;
     return true;
   }
@@ -51,12 +52,12 @@ bool AbslParseFlag(absl::string_view text,
 
 std::string AbslUnparseFlag(LiteRtGoogleTensorOptionsTruncationType options) {
   switch (options) {
-    case kLiteRtGoogleTensorFloatTruncationTypeUnspecified:
-      return "unspecified";
+    case kLiteRtGoogleTensorFloatTruncationTypeAuto:
+      return "auto";
     case kLiteRtGoogleTensorFloatTruncationTypeNoTruncation:
       return "no_truncation";
     case kLiteRtGoogleTensorFloatTruncationTypeBfloat16:
-      return "bf16";
+      return "bfloat16";
     case kLiteRtGoogleTensorFloatTruncationTypeHalf:
       return "half";
   }
@@ -101,15 +102,12 @@ std::string AbslUnparseFlag(
 
 ABSL_FLAG(LiteRtGoogleTensorOptionsTruncationType,
           google_tensor_truncation_type,
-          kLiteRtGoogleTensorFloatTruncationTypeUnspecified,
+          kLiteRtGoogleTensorFloatTruncationTypeAuto,
           "Float truncation type for Google Tensor.");
 
 ABSL_FLAG(bool, google_tensor_int64_to_int32, false,
           "Whether to truncate int64 to int32.");
 
-ABSL_FLAG(std::string, google_tensor_output_dir, "",
-          "Output directory for Google Tensor.");
-
 ABSL_FLAG(bool, google_tensor_dump_op_timings, false,
           "Whether to dump op timings.");
 
@@ -137,7 +135,6 @@ Expected<GoogleTensorOptions> GoogleTensorOptionsFromFlags() {
       absl::GetFlag(FLAGS_google_tensor_truncation_type));
   options.SetInt64ToInt32Truncation(
       absl::GetFlag(FLAGS_google_tensor_int64_to_int32));
-  options.SetOutputDir(absl::GetFlag(FLAGS_google_tensor_output_dir));
   options.SetDumpOpTimings(absl::GetFlag(FLAGS_google_tensor_dump_op_timings));
   options.SetEnableLargeModelSupport(
       absl::GetFlag(FLAGS_google_tensor_enable_large_model_support));

litert/tools/flags/vendors/google_tensor_flags.h

@@ -37,8 +37,6 @@ std::string AbslUnparseFlag(LiteRtGoogleTensorOptionsTruncationType options);
 
 ABSL_DECLARE_FLAG(bool, google_tensor_int64_to_int32);
 
-ABSL_DECLARE_FLAG(std::string, google_tensor_output_dir);
-
 ABSL_DECLARE_FLAG(bool, google_tensor_dump_op_timings);
 
 ABSL_DECLARE_FLAG(bool, google_tensor_enable_large_model_support);

litert/tools/flags/vendors/google_tensor_flags_test.cc

@@ -18,6 +18,7 @@
 
 #include <gtest/gtest.h>
 #include "litert/c/options/litert_google_tensor_options.h"
+#include "litert/c/options/litert_google_tensor_options_type.h"
 
 namespace litert::google_tensor {
 namespace {
@@ -34,9 +35,9 @@ TEST(TruncationTypeFlagTest, Parse) {
   LiteRtGoogleTensorOptionsTruncationType value;
 
   {
-    static constexpr absl::string_view kLevel = "unspecified";
+    static constexpr absl::string_view kLevel = "auto";
     static constexpr LiteRtGoogleTensorOptionsTruncationType kLevelEnum =
-        kLiteRtGoogleTensorFloatTruncationTypeUnspecified;
+        kLiteRtGoogleTensorFloatTruncationTypeAuto;
     EXPECT_TRUE(AbslParseFlag(kLevel, &value, &error));
     EXPECT_EQ(value, kLevelEnum);
     EXPECT_EQ(kLevel, AbslUnparseFlag(value));
@@ -52,7 +53,7 @@ TEST(TruncationTypeFlagTest, Parse) {
   }
 
   {
-    static constexpr absl::string_view kLevel = "bf16";
+    static constexpr absl::string_view kLevel = "bfloat16";
     static constexpr LiteRtGoogleTensorOptionsTruncationType kLevelEnum =
         kLiteRtGoogleTensorFloatTruncationTypeBfloat16;
     EXPECT_TRUE(AbslParseFlag(kLevel, &value, &error));