mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-11-03 18:00:55 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

'fill_dptr' instead of 'zero_dptr'

Browse Source
This commit is contained in:
Steven Robertson
2011-10-27 10:35:01 -04:00
parent 3c1dac530b
commit 1faffa1d14
2 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
cuburn/code/util.py
View File

@ -90,10 +90,10 @@ uint32_t trunca(float f) {
}
__global__
void zero_dptr(float* dptr, int size) {
void fill_dptr(uint32_t* dptr, int size, uint32_t value) {
int i = (gridDim.x * blockIdx.y + blockIdx.x) * blockDim.x + threadIdx.x;
if (i < size) {
dptr[i] = 0.0f;
dptr[i] = value;
}
}
@ -185,14 +185,21 @@ float3 hsv2rgb(float3 hsv) {
"""
@staticmethod
def zero_dptr(mod, dptr, size, stream=None):
def fill_dptr(mod, dptr, size, stream=None, value=np.uint32(0)):
"""
A memory zeroer which can be embedded in a stream. Size is the
number of 4-byte words in the pointer.
A memory zeroer which can be embedded in a stream, unlike the various
memset routines. Size is the number of 4-byte words in the pointer;
value is the word to fill it with. If value is not an np.uint32, it
will be coerced to a buffer and the first four bytes taken.
"""
zero = mod.get_function("zero_dptr")
fill = mod.get_function("fill_dptr")
if not isinstance(value, np.uint32):
if isinstance(value, int):
value = np.uint32(value)
else:
value = np.frombuffer(buffer(value), np.uint32)[0]
blocks = int(np.ceil(np.sqrt(size / 1024 + 1)))
zero(dptr, np.int32(size), stream=stream,
fill(dptr, np.int32(size), value, stream=stream,
block=(1024, 1, 1), grid=(blocks, blocks, 1))