[CI] explicitly run unit tests (#54)

This commit is contained in:
Shintaro Iwasaki
2022-08-12 13:39:04 -07:00
committed by GitHub
parent 2ba9a83465
commit d5856435d7
2 changed files with 10 additions and 1 deletions

View File

@@ -48,3 +48,12 @@ jobs:
alias python='python3'
cd python
pip3 install -e '.[tests]'
- name: Run tests
run: |
cd python
LIT_TEST_DIR="build/$(ls build)/test"
if [ ! -d "$LIT_TEST_DIR" ]; then
echo "Not found `$LIT_TEST_DIR`. Did you change an installation method?" ; exit -1
fi
lit -v "$LIT_TEST_DIR"

View File

@@ -59,7 +59,7 @@ func @test_combine_select_masked_load_pattern(%ptr: tensor<8x!tt.ptr<f32>>, %con
// CHECK-LABEL: @test_combine_broadcast_constant_pattern
func @test_combine_broadcast_constant_pattern(%cst : f32) -> tensor<8x2xf32> {
// CHECK: %[[cst:.*]] = arith.constant dense<0.000000e+00> : tensor<8x2xf32>
// CHECK: %[[cst:.*]] = arith.constant dense<1.000000e+00> : tensor<8x2xf32>
%const = arith.constant dense<1.0> : tensor<8xf32>
%bst_out = tt.broadcast %const : (tensor<8xf32>) -> tensor<8x2xf32>