mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix SIMD bit shifts

Browse Source
This commit is contained in:
Seo Sanghyeon 2014-01-30 00:28:29 +09:00
parent b3d10f4383
commit 5f68142d0e
2 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/librustc/middle/trans/base.rs
View File

@ -30,7 +30,7 @@ use driver::session;
use driver::session::Session;
use driver::driver::{CrateAnalysis, CrateTranslation};
use lib::llvm::{ModuleRef, ValueRef, BasicBlockRef};
use lib::llvm::{llvm, True};
use lib::llvm::{llvm, True, Vector};
use lib;
use metadata::common::LinkMeta;
use metadata::{csearch, encoder};
@ -827,8 +827,10 @@ pub fn cast_shift_rhs(op: ast::BinOp,
// Shifts may have any size int on the rhs
unsafe {
if ast_util::is_shift_binop(op) {
let rhs_llty = val_ty(rhs);
let lhs_llty = val_ty(lhs);
let mut rhs_llty = val_ty(rhs);
let mut lhs_llty = val_ty(lhs);
if rhs_llty.kind() == Vector { rhs_llty = rhs_llty.element_type() }
if lhs_llty.kind() == Vector { lhs_llty = lhs_llty.element_type() }
let rhs_sz = llvm::LLVMGetIntTypeWidth(rhs_llty.to_ref());
let lhs_sz = llvm::LLVMGetIntTypeWidth(lhs_llty.to_ref());
if lhs_sz < rhs_sz {

10
src/test/run-pass/simd-binop.rs
View File

@ -10,7 +10,7 @@
#[allow(experimental)];
use std::unstable::simd::{i32x4, f32x4};
use std::unstable::simd::{i32x4, f32x4, u32x4};
fn test_int(e: i32) -> i32 {
let v = i32x4(e, 0i32, 0i32, 0i32);
@ -24,7 +24,15 @@ fn test_float(e: f32) -> f32 {
e2
}
pub fn test_shift(e: u32) -> u32 {
let v = u32x4(e, 0u32, 0u32, 0u32);
let one = u32x4(1u32, 0u32, 0u32, 0u32);
let u32x4(e2, _, _, _) = v << one >> one;
e2
}
pub fn main() {
assert_eq!(test_int(3i32), 9i32);
assert_eq!(test_float(3f32), 9f32);
assert_eq!(test_shift(3u32), 3u32);
}