shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mlir] Use dyn_cast instead of cast in MathToVCIX conversion (#134047)

Browse Source 
Fixes #131093.
This commit is contained in:
Longsheng Mou 2025-04-02 18:30:42 +08:00 committed by GitHub
parent 2426ac647f
commit 7d441d9892
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mlir/test/Conversion/MathToVCIX/math-to-vcix.mlir
View File

@ -191,3 +191,14 @@ func.func @log_fixed(%a: vector<8 x f32>, %rvl: i64) -> vector<8 x f32> {
%res = math.log %a : vector<8 x f32>
return %res : vector<8 x f32>
}
// -----
// Ensure this case exit gracefully
// CHECK-LABEL: func.func @no_vector_type
// CHECK: math.cos
func.func @no_vector_type(%arg0: f32) -> f32 {
%0 = math.cos %arg0 : f32
return %0 : f32
}

2
mlir/test/lib/Conversion/MathToVCIX/TestMathToVCIXConversion.cpp
View File

@ -25,7 +25,7 @@ namespace {
/// according to LLVM's encoding:
/// https://lists.llvm.org/pipermail/llvm-dev/2020-October/145850.html
static std::pair<unsigned, VectorType> legalizeVectorType(const Type &type) {
VectorType vt = cast<VectorType>(type);
VectorType vt = dyn_cast<VectorType>(type);
// To simplify test pass, avoid multi-dimensional vectors.
if (!vt || vt.getRank() != 1)
return {0, nullptr};