Sha256: 5279847a52c0454cfe512a46007a65a7721ff67dfd63426127e40fbbc2858ef0
Contents?: true
Size: 1.1 KB
Versions: 19
Compression:
Stored size: 1.1 KB
Contents
static void
<%=c_iter%>(cumo_na_loop_t *const lp)
{
size_t i;
CUMO_BIT_DIGIT *a1, x=0;
size_t p1;
ssize_t s1;
size_t *idx1;
VALUE y;
CUMO_INIT_COUNTER(lp, i);
CUMO_INIT_PTR_BIT_IDX(lp, 0, a1, p1, s1, idx1);
CUMO_SHOW_SYNCHRONIZE_WARNING_ONCE("<%=name%>", "<%=type_name%>");
cumo_cuda_runtime_check_status(cudaDeviceSynchronize());
if (idx1) {
for (; i--;) {
CUMO_LOAD_BIT(a1, p1+*idx1, x); idx1++;
y = m_data_to_num(x);
rb_yield(y);
}
} else {
for (; i--;) {
CUMO_LOAD_BIT(a1, p1, x); p1+=s1;
y = m_data_to_num(x);
rb_yield(y);
}
}
}
/*
Calls the given block once for each element in self,
passing that element as a parameter.
@overload <%=name%>
@return [Cumo::NArray] self
For a block {|x| ... }
@yield [x] x is element of NArray.
*/
static VALUE
<%=c_func(0)%>(VALUE self)
{
cumo_ndfunc_arg_in_t ain[1] = {{Qnil,0}};
cumo_ndfunc_t ndf = {<%=c_iter%>, CUMO_FULL_LOOP_NIP, 1,0, ain,0};
cumo_na_ndloop(&ndf, 1, self);
return self;
}
Version data entries
19 entries across 19 versions & 1 rubygems