diff --git a/tests/lint/rust_format.sh b/tests/lint/rust_format.sh new file mode 100755 index 000000000000..10c8feec1fcf --- /dev/null +++ b/tests/lint/rust_format.sh @@ -0,0 +1,35 @@ +#!/bin/bash +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you 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. + +TVM_HOME="$(git rev-parse --show-toplevel)" +RUST_DIR="$TVM_HOME/rust" + +if [[ "$1" == "-i" ]]; then + INPLACE_FORMAT=1 + shift 1 +else + INPLACE_FORMAT=0 +fi + +cd $RUST_DIR + +if [[ ${INPLACE_FORMAT} -eq 1 ]]; then + cargo fmt +else + cargo fmt -- --check +fi diff --git a/tests/scripts/task_lint.sh b/tests/scripts/task_lint.sh index 12b32709a392..2889c3a94f11 100755 --- a/tests/scripts/task_lint.sh +++ b/tests/scripts/task_lint.sh @@ -39,6 +39,9 @@ tests/lint/cpplint.sh echo "clang-format check..." tests/lint/clang_format.sh +echo "Rust check..." +tests/lint/rust_format.sh + echo "black check..." tests/lint/python_format.sh diff --git a/tests/scripts/task_rust.sh b/tests/scripts/task_rust.sh index 4b34b6cf8db4..5cc1dc0503f7 100755 --- a/tests/scripts/task_rust.sh +++ b/tests/scripts/task_rust.sh @@ -43,9 +43,6 @@ echo "Using TVM_CARGO_VERSION=$TVM_CARGO_VERSION" export TVM_BIND_THREADS=0 export OMP_NUM_THREADS=1 -cd $RUST_DIR -cargo fmt -- --check - # First we test tvm-sys the core Rust bindings. cd $RUST_DIR/tvm-sys # First we test w/o the bindings feature on.