Skip to content

Commit

Permalink
Merge pull request rust-lang#25 from 6A/master
Browse files Browse the repository at this point in the history
AddressSpace everywhere (fixes rust-lang#24)
  • Loading branch information
TheDan64 authored Oct 27, 2017
2 parents b75c231 + dabea27 commit d93253d
Show file tree
Hide file tree
Showing 12 changed files with 72 additions and 65 deletions.
27 changes: 27 additions & 0 deletions src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,33 @@ pub fn shutdown_llvm() {
}
}

/// Defines the address space in which a global will be inserted.
///
/// # Remarks
/// See also: http://llvm.org/doxygen/NVPTXBaseInfo_8h_source.html
#[repr(u32)]
#[derive(Debug, PartialEq, Eq, Copy, Clone)]
pub enum AddressSpace {
Generic = 0,
Global = 1,
Shared = 3,
Const = 4,
Local = 5
}

impl From<u32> for AddressSpace {
fn from(val: u32) -> Self {
match val {
0 => AddressSpace::Generic,
1 => AddressSpace::Global,
2 => AddressSpace::Shared,
3 => AddressSpace::Const,
4 => AddressSpace::Local,
_ => unreachable!("Invalid value for AddressSpace"),
}
}
}

// REVIEW: Maybe this belongs in some sort of prelude?
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum IntPredicate {
Expand Down
28 changes: 1 addition & 27 deletions src/module.rs
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ use std::mem::{forget, uninitialized, zeroed};
use std::path::Path;
use std::slice::from_raw_parts;

use OptimizationLevel;
use {AddressSpace, OptimizationLevel};
use context::{Context, ContextRef};
use data_layout::DataLayout;
use execution_engine::ExecutionEngine;
Expand Down Expand Up @@ -41,32 +41,6 @@ pub enum Linkage {
WeakODRLinkage,
}

/// Defines the address space in which a global will be inserted.
///
/// # Remarks
/// See also: http://llvm.org/doxygen/NVPTXBaseInfo_8h_source.html
#[derive(Debug, PartialEq, Eq, Copy, Clone)]
pub enum AddressSpace {
Generic = 0,
Global = 1,
Shared = 3,
Const = 4,
Local = 5
}

impl From<u32> for AddressSpace {
fn from(val: u32) -> Self {
match val {
0 => AddressSpace::Generic,
1 => AddressSpace::Global,
2 => AddressSpace::Shared,
3 => AddressSpace::Const,
4 => AddressSpace::Local,
_ => unreachable!("Invalid value for AddressSpace"),
}
}
}

impl Linkage {
pub(crate) fn new(linkage: LLVMLinkage) -> Self {
match linkage {
Expand Down
3 changes: 2 additions & 1 deletion src/types/array_type.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ use llvm_sys::prelude::{LLVMTypeRef, LLVMValueRef};

use std::ffi::CStr;

use AddressSpace;
use context::ContextRef;
use types::traits::AsTypeRef;
use types::{Type, BasicType, PointerType, FunctionType};
Expand Down Expand Up @@ -36,7 +37,7 @@ impl ArrayType {
None
}

pub fn ptr_type(&self, address_space: u32) -> PointerType {
pub fn ptr_type(&self, address_space: AddressSpace) -> PointerType {
self.array_type.ptr_type(address_space)
}

Expand Down
3 changes: 2 additions & 1 deletion src/types/float_type.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ use llvm_sys::prelude::LLVMTypeRef;

use std::ffi::CStr;

use AddressSpace;
use context::ContextRef;
use types::traits::AsTypeRef;
use types::{Type, PointerType, FunctionType, BasicType, ArrayType, VectorType};
Expand Down Expand Up @@ -77,7 +78,7 @@ impl FloatType {
self.float_type.get_context()
}

pub fn ptr_type(&self, address_space: u32) -> PointerType {
pub fn ptr_type(&self, address_space: AddressSpace) -> PointerType {
self.float_type.ptr_type(address_space)
}

Expand Down
3 changes: 2 additions & 1 deletion src/types/int_type.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ use llvm_sys::prelude::LLVMTypeRef;

use std::ffi::CStr;

use AddressSpace;
use context::ContextRef;
use types::traits::AsTypeRef;
use types::{Type, ArrayType, BasicType, VectorType, PointerType, FunctionType};
Expand Down Expand Up @@ -296,7 +297,7 @@ impl IntType {
self.int_type.size_of()
}

pub fn ptr_type(&self, address_space: u32) -> PointerType {
pub fn ptr_type(&self, address_space: AddressSpace) -> PointerType {
self.int_type.ptr_type(address_space)
}

Expand Down
5 changes: 3 additions & 2 deletions src/types/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@ use llvm_sys::prelude::{LLVMTypeRef, LLVMValueRef};
use std::ffi::CStr;
use std::fmt;

use AddressSpace;
use context::{Context, ContextRef};
use values::{IntValue, PointerValue};

Expand Down Expand Up @@ -62,9 +63,9 @@ impl Type {
PointerValue::new(ptr_type)
}

fn ptr_type(&self, address_space: u32) -> PointerType {
fn ptr_type(&self, address_space: AddressSpace) -> PointerType {
let ptr_type = unsafe {
LLVMPointerType(self.type_, address_space)
LLVMPointerType(self.type_, address_space as u32)
};

PointerType::new(ptr_type)
Expand Down
4 changes: 2 additions & 2 deletions src/types/ptr_type.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@ use llvm_sys::prelude::LLVMTypeRef;

use std::ffi::CStr;

use AddressSpace;
use context::ContextRef;
use module::AddressSpace;
use types::traits::AsTypeRef;
use types::{Type, BasicType, ArrayType, FunctionType, VectorType};
use values::{PointerValue, IntValue};
Expand All @@ -31,7 +31,7 @@ impl PointerType {
self.ptr_type.size_of()
}

pub fn ptr_type(&self, address_space: u32) -> PointerType {
pub fn ptr_type(&self, address_space: AddressSpace) -> PointerType {
self.ptr_type.ptr_type(address_space)
}

Expand Down
3 changes: 2 additions & 1 deletion src/types/struct_type.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ use llvm_sys::prelude::{LLVMTypeRef, LLVMValueRef};
use std::ffi::CStr;
use std::mem::forget;

use AddressSpace;
use context::ContextRef;
use types::traits::AsTypeRef;
use types::{Type, BasicType, BasicTypeEnum, ArrayType, PointerType, FunctionType, VectorType};
Expand Down Expand Up @@ -115,7 +116,7 @@ impl StructType {
Some(c_str)
}

pub fn ptr_type(&self, address_space: u32) -> PointerType {
pub fn ptr_type(&self, address_space: AddressSpace) -> PointerType {
self.struct_type.ptr_type(address_space)
}

Expand Down
3 changes: 2 additions & 1 deletion src/types/void_type.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
use llvm_sys::core::LLVMVoidType;
use llvm_sys::prelude::LLVMTypeRef;

use AddressSpace;
use context::ContextRef;
use types::traits::AsTypeRef;
use types::{Type, BasicType, FunctionType, PointerType};
Expand Down Expand Up @@ -31,7 +32,7 @@ impl VoidType {
self.void_type.get_context()
}

pub fn ptr_type(&self, address_space: u32) -> PointerType {
pub fn ptr_type(&self, address_space: AddressSpace) -> PointerType {
self.void_type.ptr_type(address_space)
}

Expand Down
4 changes: 2 additions & 2 deletions tests/test_builder.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
extern crate inkwell;

use self::inkwell::OptimizationLevel;
use self::inkwell::{AddressSpace, OptimizationLevel};
use self::inkwell::context::Context;
use self::inkwell::builder::Builder;
use self::inkwell::targets::{InitializationConfig, Target};
Expand Down Expand Up @@ -56,7 +56,7 @@ fn test_null_checked_ptr_ops() {
// }

let i8_type = context.i8_type();
let i8_ptr_type = i8_type.ptr_type(0);
let i8_ptr_type = i8_type.ptr_type(AddressSpace::Generic);
let i64_type = context.i64_type();
let fn_type = i8_type.fn_type(&[&i8_ptr_type], false);
let neg_one = i8_type.const_all_ones();
Expand Down
41 changes: 21 additions & 20 deletions tests/test_types.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ extern crate inkwell;

use std::ffi::CString;

use self::inkwell::AddressSpace;
use self::inkwell::context::Context;
use self::inkwell::types::{FloatType, IntType, StructType, VoidType};

Expand Down Expand Up @@ -156,26 +157,26 @@ fn test_sized_types() {
assert!(!fn_type3.is_sized());
assert!(!fn_type4.is_sized());

assert!(void_type.ptr_type(0).is_sized());
assert!(bool_type.ptr_type(0).is_sized());
assert!(i8_type.ptr_type(0).is_sized());
assert!(i16_type.ptr_type(0).is_sized());
assert!(i32_type.ptr_type(0).is_sized());
assert!(i64_type.ptr_type(0).is_sized());
assert!(i128_type.ptr_type(0).is_sized());
assert!(f16_type.ptr_type(0).is_sized());
assert!(f32_type.ptr_type(0).is_sized());
assert!(f64_type.ptr_type(0).is_sized());
assert!(f128_type.ptr_type(0).is_sized());
assert!(ppc_f128_type.ptr_type(0).is_sized());
assert!(struct_type.ptr_type(0).is_sized());
assert!(struct_type2.ptr_type(0).is_sized());
assert!(struct_type3.ptr_type(0).is_sized());
assert!(struct_type4.ptr_type(0).is_sized());
assert!(opaque_struct_type.ptr_type(0).is_sized());
assert!(void_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(bool_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(i8_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(i16_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(i32_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(i64_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(i128_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(f16_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(f32_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(f64_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(f128_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(ppc_f128_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(struct_type.ptr_type(AddressSpace::Generic).is_sized());
assert!(struct_type2.ptr_type(AddressSpace::Generic).is_sized());
assert!(struct_type3.ptr_type(AddressSpace::Generic).is_sized());
assert!(struct_type4.ptr_type(AddressSpace::Generic).is_sized());
assert!(opaque_struct_type.ptr_type(AddressSpace::Generic).is_sized());

// REVIEW: You can't have array of void right?
assert!(void_type.ptr_type(0).array_type(42).is_sized());
assert!(void_type.ptr_type(AddressSpace::Generic).array_type(42).is_sized());
assert!(bool_type.array_type(42).is_sized());
assert!(i8_type.array_type(42).is_sized());
assert!(i16_type.array_type(42).is_sized());
Expand All @@ -194,7 +195,7 @@ fn test_sized_types() {
assert!(!opaque_struct_type.array_type(0).is_sized());

// REVIEW: You can't have vec of void right?
assert!(void_type.ptr_type(0).vec_type(42).is_sized());
assert!(void_type.ptr_type(AddressSpace::Generic).vec_type(42).is_sized());
assert!(bool_type.vec_type(42).is_sized());
assert!(i8_type.vec_type(42).is_sized());
assert!(i16_type.vec_type(42).is_sized());
Expand Down Expand Up @@ -228,7 +229,7 @@ fn test_const_null() {
let f64_type = context.f64_type();
let f128_type = context.f128_type();
let struct_type = context.struct_type(&[&i8_type, &f128_type], false);
let ptr_type = f64_type.ptr_type(0);
let ptr_type = f64_type.ptr_type(AddressSpace::Generic);
let vec_type = f64_type.vec_type(42);
let array_type = f64_type.array_type(42);

Expand Down
13 changes: 6 additions & 7 deletions tests/test_values.rs
Original file line number Diff line number Diff line change
@@ -1,8 +1,7 @@
extern crate inkwell;

use self::inkwell::{DLLStorageClass, FloatPredicate, GlobalVisibility, ThreadLocalMode};
use self::inkwell::{DLLStorageClass, FloatPredicate, GlobalVisibility, ThreadLocalMode, AddressSpace};
use self::inkwell::context::Context;
use self::inkwell::module::AddressSpace::*;
use self::inkwell::module::Linkage::*;
use self::inkwell::types::{StructType, VectorType};
use self::inkwell::values::InstructionOpcode::*;
Expand Down Expand Up @@ -32,7 +31,7 @@ fn test_instructions() {
let void_type = context.void_type();
let i64_type = context.i64_type();
let f32_type = context.f32_type();
let f32_ptr_type = f32_type.ptr_type(0);
let f32_ptr_type = f32_type.ptr_type(AddressSpace::Generic);
let fn_type = void_type.fn_type(&[&f32_ptr_type], false);

let function = module.add_function("free_f32", &fn_type, None);
Expand Down Expand Up @@ -170,7 +169,7 @@ fn test_set_get_name() {
assert_eq!(ppc_f128_val.get_name(), &*CString::new("").unwrap());

let void_type = context.void_type();
let ptr_type = bool_type.ptr_type(0);
let ptr_type = bool_type.ptr_type(AddressSpace::Generic);
let struct_type = context.struct_type(&[&bool_type], false);
let vec_type = bool_type.vec_type(1);

Expand Down Expand Up @@ -278,7 +277,7 @@ fn test_undef() {
let f32_undef = f32_type.get_undef();
let f64_undef = f64_type.get_undef();
let f128_undef = f128_type.get_undef();
let ptr_undef = bool_type.ptr_type(0).get_undef();
let ptr_undef = bool_type.ptr_type(AddressSpace::Generic).get_undef();
let array_undef = array_type.get_undef();
let struct_undef = StructType::struct_type(&[&bool_type], false).get_undef();
let vec_undef = bool_type.vec_type(1).get_undef();
Expand Down Expand Up @@ -448,7 +447,7 @@ fn test_metadata() {
let f64_val = f64_type.const_float(0.0);
let f128_val = f128_type.const_float(0.0);
// let ppc_f128_val = ppc_f128_type.const_float(0.0);
let ptr_val = bool_type.ptr_type(0).const_null();
let ptr_val = bool_type.ptr_type(AddressSpace::Generic).const_null();
let array_val = array_type.const_array(&[&f64_val]);
let struct_val = context.const_struct(&[&i8_val, &f128_val], false);
let vec_val = VectorType::const_vector(&[&i8_val]);
Expand Down Expand Up @@ -791,7 +790,7 @@ fn test_globals() {

assert!(global.get_thread_local_mode().is_none());

let global2 = module.add_global(&i8_type, Some(Const), "my_global2");
let global2 = module.add_global(&i8_type, Some(AddressSpace::Const), "my_global2");

assert_eq!(global2.get_previous_global().unwrap(), global);
assert_eq!(global.get_next_global().unwrap(), global2);
Expand Down

0 comments on commit d93253d

Please sign in to comment.