Update on "Extending _get_bytecode_version to support flatbuffers format"

pavithranrao · pavithranrao · commit d00bb33e1efc · 2022-03-31T20:51:21.000-07:00
Extending `_get_bytecode_version` to support flatbuffers. Differential Revision: [D34900498](https://our.internmc.facebook.com/intern/diff/D34900498/) **NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D34900498/)! [ghstack-poisoned]
diff --git a/test/cpp/jit/test_flatbuffer.cpp b/test/cpp/jit/test_flatbuffer.cpp
@@ -257,7 +257,7 @@ TEST(FlatbufferTest, GetByteCodeVersion) {
       return input + 1
   )");
   std::stringstream ss;
-  m._save_for_mobile(ss, {}, false, true);
+  m._save_for_mobile(ss, {}, false, /*use_flatbuffer=*/true);
   auto version = _get_model_bytecode_version(ss);
   AT_ASSERT(version == caffe2::serialize::kProducedBytecodeVersion);
   ss.seekg(0, ss.beg);
