Make mem.set use llvm.memset.p0i8.iXX

This commit is contained in:
gingerBill
2019-12-31 14:09:51 +00:00
parent c407687a4c
commit 13107628f8
2 changed files with 32 additions and 18 deletions
+12 -1
View File
@@ -12,7 +12,18 @@ swap :: proc{swap16, swap32, swap64};
set :: proc "contextless" (data: rawptr, value: byte, len: int) -> rawptr { set :: proc "contextless" (data: rawptr, value: byte, len: int) -> rawptr {
return runtime.memset(data, i32(value), len); foreign _ {
when size_of(rawptr) == 8 {
@(link_name="llvm.memset.p0i8.i64")
llvm_memset :: proc(dst: rawptr, val: byte, len: int, align: i32, is_volatile: bool) ---;
} else {
@(link_name="llvm.memset.p0i8.i32")
llvm_memset :: proc(dst: rawptr, val: byte, len: int, align: i32, is_volatile: bool) ---;
}
}
llvm_memset(data, value, len, 1, false);
return data;
} }
zero :: inline proc "contextless" (data: rawptr, len: int) -> rawptr { zero :: inline proc "contextless" (data: rawptr, len: int) -> rawptr {
return set(data, 0, len); return set(data, 0, len);
+20 -17
View File
@@ -3,25 +3,28 @@ package runtime
@(link_name="memset") @(link_name="memset")
memset :: proc "c" (ptr: rawptr, val: i32, len: int) -> rawptr { memset :: proc "c" (ptr: rawptr, val: i32, len: int) -> rawptr {
when false { when false {
b := byte(val); b := byte(val);
p_start := uintptr(ptr); p_start := uintptr(ptr);
p_end := p + uintptr(max(len, 0)); p_end := p + uintptr(max(len, 0));
for p := p_start; p < p_end; p += 1 { for p := p_start; p < p_end; p += 1 {
(^byte)(p)^ = b; (^byte)(p)^ = b;
} }
return ptr; return ptr;
} else {
when size_of(rawptr) == 8 {
@(link_name="llvm.memset.p0i8.i64")
llvm_memset :: proc(dst: rawptr, val: byte, len: int, align: i32, is_volatile: bool) ---;
} else { } else {
@(link_name="llvm.memset.p0i8.i32") foreign _ {
llvm_memset :: proc(dst: rawptr, val: byte, len: int, align: i32, is_volatile: bool) ---; when size_of(rawptr) == 8 {
} @(link_name="llvm.memset.p0i8.i64")
llvm_memset :: proc(dst: rawptr, val: byte, len: int, align: i32, is_volatile: bool) ---;
} else {
@(link_name="llvm.memset.p0i8.i32")
llvm_memset :: proc(dst: rawptr, val: byte, len: int, align: i32, is_volatile: bool) ---;
}
return llvm_memset(ptr, byte(val), len, 1, false); }
} llvm_memset(ptr, byte(val), len, 1, false);
return ptr;
}
} }