diff --git a/binaries/CMakeLists.txt b/binaries/CMakeLists.txt index 3405b1defb5..fa198928e69 100644 --- a/binaries/CMakeLists.txt +++ b/binaries/CMakeLists.txt @@ -24,17 +24,6 @@ if(BUILD_TEST) target_link_libraries(core_overhead_benchmark benchmark) endif() -if(USE_CUDA) - caffe2_binary_target("inspect_gpu.cc") - target_link_libraries(inspect_gpu ${CUDA_LIBRARIES}) - -endif() - -if(USE_ROCM) - caffe2_hip_binary_target("hip/inspect_gpu.cc") - -endif() - caffe2_binary_target("dump_operator_names.cc") caffe2_binary_target("optimize_for_mobile.cc") diff --git a/binaries/inspect_gpu.cc b/binaries/inspect_gpu.cc deleted file mode 100644 index b4563cc8c31..00000000000 --- a/binaries/inspect_gpu.cc +++ /dev/null @@ -1,58 +0,0 @@ -/** - * Copyright (c) 2016-present, Facebook, Inc. - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -#include - -#include -#include - -#include "c10/util/Flags.h" -#include "caffe2/core/common_gpu.h" -#include "caffe2/core/init.h" -#include "caffe2/core/logging.h" - -using std::vector; - -C10_DECLARE_int(caffe2_log_level); - -int main(int argc, char** argv) { - caffe2::GlobalInit(&argc, &argv); - c10::SetUsageMessage( - "Inspects the GPUs on the current machine and prints out their details " - "provided by cuda."); - - int gpu_count; - CUDA_ENFORCE(cudaGetDeviceCount(&gpu_count)); - for (int i = 0; i < gpu_count; ++i) { - LOG(INFO) << "Querying device ID = " << i; - caffe2::DeviceQuery(i); - } - - vector > access_pattern; - CAFFE_ENFORCE(caffe2::GetCudaPeerAccessPattern(&access_pattern)); - - std::stringstream sstream; - // Find topology - for (int i = 0; i < gpu_count; ++i) { - for (int j = 0; j < gpu_count; ++j) { - sstream << (access_pattern[i][j] ? "+" : "-") << " "; - } - sstream << std::endl; - } - LOG(INFO) << "Access pattern: " << std::endl << sstream.str(); - - return 0; -}