Skip to content

Commit

Permalink
Merge remote-tracking branch 'upstream/master' into develop-upstream-…
Browse files Browse the repository at this point in the history
…sync-220711 [merge conflicts resolved]
  • Loading branch information
rsanthanam-amd committed Jul 11, 2022
1 parent bb08c1f commit 45d7533
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 11 deletions.
4 changes: 0 additions & 4 deletions tensorflow/core/tfrt/saved_model/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -35,10 +35,6 @@ cc_library(
tags = ["no_oss"],
deps = [
"@com_google_absl//absl/container:flat_hash_map",
<<<<<<< HEAD
# "@com_google_absl//absl/log:check",
=======
>>>>>>> upstream/master
"@com_google_absl//absl/strings",
"@com_google_absl//absl/synchronization",
"@com_google_absl//absl/types:span",
Expand Down
9 changes: 2 additions & 7 deletions tensorflow/stream_executor/multi_platform_manager.cc
Original file line number Diff line number Diff line change
Expand Up @@ -127,24 +127,19 @@ port::StatusOr<Platform*> MultiPlatformManagerImpl::PlatformWithId(
port::StatusOr<Platform*> MultiPlatformManagerImpl::PlatformWithName(
absl::string_view target, bool initialize_platform) {
absl::MutexLock lock(&mu_);
<<<<<<< HEAD
Platform * platform = 0;
if(target == "cuda" || target == "CUDA") {
port::StatusOr<Platform*> lookup = LookupByNameLocked("cuda");
if(lookup.ok())
platform = lookup.value();
else {
SE_ASSIGN_OR_RETURN(platform, LookupByNameLocked("rocm"));
TF_ASSIGN_OR_RETURN(platform, LookupByNameLocked("rocm"));
}
} else {
if(target == "cuda_only")
target = "cuda";
SE_ASSIGN_OR_RETURN(platform, LookupByNameLocked(target));
TF_ASSIGN_OR_RETURN(platform, LookupByNameLocked(target));
}
=======

TF_ASSIGN_OR_RETURN(Platform * platform, LookupByNameLocked(target));
>>>>>>> upstream/master
if (initialize_platform && !platform->Initialized()) {
TF_RETURN_IF_ERROR(platform->Initialize({}));
}
Expand Down

0 comments on commit 45d7533

Please sign in to comment.