Apply similar changes to gcs

Author: oandreeva-nv

src/filesystem/implementations/gcs.h

@@ -384,9 +384,23 @@ GCSFileSystem::LocalizePath(
         "GCS file localization not yet implemented " + path);
   }
 
+  // Create a local directory for s3 model store.
+  // If `mount_dir` or ENV variable are not set,
+  // creates a temporary directory under `/tmp` with the format: "folderXXXXXX".
+  // Otherwise, will create a folder under specified directory with the name
+  // indicated in path (i.e. everything after the last encounter of `/`).
+  const char* env_mount_dir = std::getenv("TRITON_GCS_MOUNT_DIRECTORY");
   std::string tmp_folder;
-  RETURN_IF_ERROR(
-      triton::core::MakeTemporaryDirectory(FileSystemType::LOCAL, &tmp_folder));
+  if (mount_dir.empty() && env_mount_dir == nullptr) {
+    RETURN_IF_ERROR(triton::core::MakeTemporaryDirectory(
+        FileSystemType::LOCAL, &tmp_folder));
+  } else {
+    tmp_folder = mount_dir.empty() ? std::string(env_mount_dir) : mount_dir;
+    tmp_folder =
+        JoinPath({tmp_folder, path.substr(path.find_last_of('/') + 1)});
+    RETURN_IF_ERROR(triton::core::MakeDirectory(
+        tmp_folder, true /*recursive*/, true /*allow_dir_exist*/));
+  }
 
   localized->reset(new LocalizedPath(path, tmp_folder));
 
@@ -406,7 +420,7 @@ GCSFileSystem::LocalizePath(
       std::string local_fpath =
           JoinPath({(*localized)->Path(), gcs_removed_path});
       RETURN_IF_ERROR(IsDirectory(gcs_fpath, &is_subdir));
-      if (is_subdir) {
+      if (recursive && is_subdir) {
         // Create local mirror of sub-directories
 #ifdef _WIN32
         int status = mkdir(const_cast<char*>(local_fpath.c_str()));
@@ -429,7 +443,7 @@ GCSFileSystem::LocalizePath(
              ++itr) {
           contents.insert(JoinPath({gcs_fpath, *itr}));
         }
-      } else {
+      } else if (!is_subdir) {
         // Create local copy of file
         std::string file_bucket, file_object;
         RETURN_IF_ERROR(ParsePath(gcs_fpath, &file_bucket, &file_object));