[ROCm] Add script to run multi gpu tests #40
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Copyright 2023 The TensorFlow Authors. All Rights Reserved. | |
| # | |
| # 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. | |
| # ============================================================================ | |
| # The goal of these checks is to externalize all check_contents checks that run | |
| # internally. For now, no automation will try to keep these in sync as many | |
| # METADATA files are doing different checks, and we'll have to consolidate those | |
| # files once XLA moves out of Tensorflow internally. | |
| # TODO(ddunleavy): Update this after METADATA files are consolidated. | |
| name: Check Contents | |
| on: | |
| pull_request: | |
| env: | |
| # A bit tricky here: this lets us invoke python files outside of `bazel run` | |
| # as long as the python file has imports as if it expects bazel to invoke it, | |
| # and only depends on other python files in this workspace or the standard | |
| # library. | |
| PYTHONPATH: ".." | |
| jobs: | |
| check-contents: | |
| runs-on: ubuntu-22.04 | |
| defaults: | |
| run: | |
| shell: bash | |
| timeout-minutes: 1 | |
| if: | | |
| github.event.sender.type == 'User' || | |
| contains(github.event.pull_request.body, 'FORCE_TEST_ACTIONS') | |
| steps: | |
| - name: "Checking out repository" | |
| uses: actions/checkout@e2f20e631ae6d7dd3b768f56a5d2af784dd54791 # v2.5.0 | |
| - name: "Fetch HEAD of main branch" | |
| run: git fetch origin main --depth=1 | |
| # ----------------C++ checks---------------- | |
| - name: "Check for tsl::{Status,StatusOr}" | |
| if: always() # We want to run all checks, not just til first failure | |
| run: "python3 build_tools/lint/check_contents.py | |
| --path_regex_exclusion '.github/workflows/check_contents\\.yml' | |
| --prohibited_regex 'tsl::Status(Or)?' | |
| --suppression_regex 'TENSORFLOW_STATUS_OK' | |
| --failure_message 'Use plain Status(Or) rather than tsl::Status(Or) within XLA.'" | |
| - name: "Check for tsl::Status::OK()" | |
| if: always() # We want to run all checks, not just til first failure | |
| run: "python3 build_tools/lint/check_contents.py | |
| --path_regex_exclusion '.github/workflows/check_contents\\.yml' | |
| --prohibited_regex 'tsl::Status::OK' | |
| --failure_message 'Use OkStatus instead of Status::OK.'" | |
| - name: "Check for std::call_once" | |
| if: always() # We want to run all checks, not just til first failure | |
| run: "python3 build_tools/lint/check_contents.py | |
| --path_regex_exclusion '.github/workflows/check_contents\\.yml' | |
| --prohibited_regex 'std::call_once' | |
| --failure_message 'Use absl::call_once instead of std::call_once'" | |
| - name: "Check for Abseil compatibility API" | |
| if: always() # We want to run all checks, not just til first failure | |
| run: "python3 build_tools/lint/check_contents.py | |
| --path_regex_exclusion '.github/workflows/check_contents\\.yml' | |
| --prohibited_regex '\\b(absl::(\ | |
| any|\ | |
| get|\ | |
| get_if|\ | |
| make_any|\ | |
| make_unique|\ | |
| nullopt|\ | |
| nullopt_t|\ | |
| MakeUnique|\ | |
| make_optional|\ | |
| optional|\ | |
| underlying_type_t|\ | |
| variant|\ | |
| visit\ | |
| ))\\b' | |
| --suppression_regex '(?i)non-std ok' | |
| --failure_message 'Use standard library version instead of the Abseil compatibility API.'" | |
| - name: "Use absl instead of Tensorflow/TSL functions/types" | |
| if: always() # We want to run all checks, not just til first failure | |
| run: "python3 build_tools/lint/check_contents.py | |
| --path_regex_exclusion '.github/workflows/check_contents\\.yml' | |
| --prohibited_regex '\\b(\ | |
| gtl::FlatMap|\ | |
| gtl::FlatSet|\ | |
| gtl::InlinedVector|\ | |
| gtl::optional|\ | |
| strings::Appendf|\ | |
| strings::Printf|\ | |
| strings::safe_strto64|\ | |
| strings::safe_strtof|\ | |
| strings::StrAppend|\ | |
| strings::StrCat|\ | |
| str_util::|\ | |
| tensorflow::StringPiece\ | |
| )\\b' | |
| --suppression_regex '(?i)non-absl ok' | |
| --failure_message 'Use Abseil types and functions instead of those in Tensorflow/TSL.'" | |
| # ----------------Python checks---------------- | |
| - name: "Check for python print()" | |
| if: always() # We want to run all checks, not just til first failure | |
| run: "python3 build_tools/lint/check_contents.py | |
| --path_regex '.*\\.py' | |
| --prohibited_regex '(^|\\n|\\s)print\\(.*\\)' | |
| --suppression_regex 'DISABLE_DEBUG_PRINT_CHECK' | |
| --failure_message 'Found debug print() statement left in the code. | |
| Add DISABLE_DEBUG_PRINT_CHECK to disable'" | |
| - name: "Check for mock.patch" | |
| if: always() # We want to run all checks, not just til first failure | |
| run: "python3 build_tools/lint/check_contents.py | |
| --path_regex '.*\\.py' | |
| --prohibited_regex 'mock\\.patch\\(' | |
| --failure_message 'Please use match.patch.object instead of mock.patch.'" | |