squiid/rust
1
0
Fork 0
Code Issues Pull requests Projects Activity

Generate LLVM SIMD vector types

Browse source
This commit is contained in:
Seo Sanghyeon 2013-05-08 02:09:19 +09:00
parent 784e8369ca
commit 0ed4495ac4
2 changed files with 55 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
src/librustc/middle/trans/type_of.rs
View file

@ -155,9 +155,15 @@ pub fn sizing_type_of(cx: @CrateContext, t: ty::t) -> TypeRef {
} }
ty::ty_struct(did, _) => { ty::ty_struct(did, _) => {
let repr = adt::represent_type(cx, t); if ty::type_is_simd(cx.tcx, t) {
let packed = ty::lookup_packed(cx.tcx, did); let et = ty::simd_type(cx.tcx, t);
T_struct(adt::sizing_fields_of(cx, repr), packed) let n = ty::simd_size(cx.tcx, t);
T_vector(type_of(cx, et), n)
} else {
let repr = adt::represent_type(cx, t);
let packed = ty::lookup_packed(cx.tcx, did);
T_struct(adt::sizing_fields_of(cx, repr), packed)
}
} }
ty::ty_self(_) | ty::ty_infer(*) | ty::ty_param(*) | ty::ty_err(*) => { ty::ty_self(_) | ty::ty_infer(*) | ty::ty_param(*) | ty::ty_err(*) => {
@ -263,14 +269,19 @@ pub fn type_of(cx: @CrateContext, t: ty::t) -> TypeRef {
} }
ty::ty_opaque_closure_ptr(_) => T_opaque_box_ptr(cx), ty::ty_opaque_closure_ptr(_) => T_opaque_box_ptr(cx),
ty::ty_struct(did, ref substs) => { ty::ty_struct(did, ref substs) => {
// Only create the named struct, but don't fill it in. We fill it if ty::type_is_simd(cx.tcx, t) {
// in *after* placing it into the type cache. This prevents let et = ty::simd_type(cx.tcx, t);
// infinite recursion with recursive struct types. let n = ty::simd_size(cx.tcx, t);
T_vector(type_of(cx, et), n)
common::T_named_struct(llvm_type_name(cx, } else {
a_struct, // Only create the named struct, but don't fill it in. We fill it
did, // in *after* placing it into the type cache. This prevents
/*bad*/ copy substs.tps)) // infinite recursion with recursive struct types.
T_named_struct(llvm_type_name(cx,
a_struct,
did,
/*bad*/ copy substs.tps))
}
} }
ty::ty_self(*) => cx.tcx.sess.unimpl(~"type_of: ty_self"), ty::ty_self(*) => cx.tcx.sess.unimpl(~"type_of: ty_self"),
ty::ty_infer(*) => cx.tcx.sess.bug(~"type_of with ty_infer"), ty::ty_infer(*) => cx.tcx.sess.bug(~"type_of with ty_infer"),
@ -289,10 +300,12 @@ pub fn type_of(cx: @CrateContext, t: ty::t) -> TypeRef {
} }
ty::ty_struct(did, _) => { ty::ty_struct(did, _) => {
let repr = adt::represent_type(cx, t); if !ty::type_is_simd(cx.tcx, t) {
let packed = ty::lookup_packed(cx.tcx, did); let repr = adt::represent_type(cx, t);
common::set_struct_body(llty, adt::fields_of(cx, repr), let packed = ty::lookup_packed(cx.tcx, did);
packed); common::set_struct_body(llty, adt::fields_of(cx, repr),
packed);
}
} }
_ => () _ => ()
} }

27
src/librustc/middle/ty.rs
View file

@ -1567,6 +1567,13 @@ pub fn type_is_sequence(ty: t) -> bool {
} }
} }
pub fn type_is_simd(cx: ctxt, ty: t) -> bool {
match get(ty).sty {
ty_struct(did, _) => lookup_simd(cx, did),
_ => false
}
}
pub fn type_is_str(ty: t) -> bool { pub fn type_is_str(ty: t) -> bool {
match get(ty).sty { match get(ty).sty {
ty_estr(_) => true, ty_estr(_) => true,
@ -1583,6 +1590,26 @@ pub fn sequence_element_type(cx: ctxt, ty: t) -> t {
} }
} }
pub fn simd_type(cx: ctxt, ty: t) -> t {
match get(ty).sty {
ty_struct(did, ref substs) => {
let fields = lookup_struct_fields(cx, did);
lookup_field_type(cx, did, fields[0].id, substs)
}
_ => fail!(~"simd_type called on invalid type")
}
}
pub fn simd_size(cx: ctxt, ty: t) -> uint {
match get(ty).sty {
ty_struct(did, _) => {
let fields = lookup_struct_fields(cx, did);
fields.len()
}
_ => fail!(~"simd_size called on invalid type")
}
}
pub fn get_element_type(ty: t, i: uint) -> t { pub fn get_element_type(ty: t, i: uint) -> t {
match get(ty).sty { match get(ty).sty {
ty_tup(ref ts) => return ts[i], ty_tup(ref ts) => return ts[i],