Generate LLVM SIMD vector types

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

View file

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

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],