Sha256: d5898b14b0bc12bab503a8d93c8b2bbda65ac73c1602b0c5b4c4f0049f5ff64b
Contents?: true
Size: 1.99 KB
Versions: 6
Compression:
Stored size: 1.99 KB
Contents
static void
<%=c_iter%>(na_loop_t *const lp)
{
size_t i;
char *p1, *p2, *p3;
ssize_t s1, s2, s3;
dtype x, y, z;
INIT_COUNTER(lp, i);
INIT_PTR(lp, 0, p1, s1);
INIT_PTR(lp, 1, p2, s2);
INIT_PTR(lp, 2, p3, s3);
if (s3==0) {
// Reduce loop
GET_DATA(p3,dtype,z);
for (; i--;) {
GET_DATA_STRIDE(p1,s1,dtype,x);
GET_DATA_STRIDE(p2,s2,dtype,y);
m_<%=method%>(x,y,z);
}
SET_DATA(p3,dtype,z);
} else {
for (; i--;) {
GET_DATA_STRIDE(p1,s1,dtype,x);
GET_DATA_STRIDE(p2,s2,dtype,y);
GET_DATA(p3,dtype,z);
m_<%=method%>(x,y,z);
SET_DATA_STRIDE(p3,s3,dtype,z);
}
}
}
static VALUE
<%=c_func%>_self(int argc, VALUE *argv, VALUE self)
{
VALUE v, reduce;
VALUE naryv[2];
ndfunc_arg_in_t ain[4] = {{cT,0},{cT,0},{sym_reduce,0},{sym_init,0}};
ndfunc_arg_out_t aout[1] = {{cT,0}};
ndfunc_t ndf = { <%=c_iter%>, STRIDE_LOOP_NIP, 4, 1, ain, aout };
if (argc < 1) {
rb_raise(rb_eArgError,"wrong number of arguments (%d for >=1)",argc);
}
// should fix below: [self.ndim,other.ndim].max or?
naryv[0] = self;
naryv[1] = argv[0];
reduce = na_reduce_dimension(argc-1, argv+1, 2, naryv);
v = na_ndloop(&ndf, 4, self, argv[0], reduce, m_<%=method%>_init);
return numo_<%=tp%>_extract(v);
}
/*
Binary <%=method%>.
@overload <%=op_map%> other
@param [Numo::NArray,Numeric] other
@return [Numo::NArray] <%=method%> of self and other.
*/
static VALUE
<%=c_func%>(int argc, VALUE *argv, VALUE self)
{
VALUE klass, v;
if (argc < 1) {
rb_raise(rb_eArgError,"wrong number of arguments (%d for >=1)",argc);
}
klass = na_upcast(CLASS_OF(self),CLASS_OF(argv[0]));
if (klass==cT) {
return <%=c_func%>_self(argc, argv, self);
} else {
v = rb_funcall(klass, id_cast, 1, self);
return rb_funcall2(v, rb_intern("<%=method%>"), argc, argv);
}
}
Version data entries
6 entries across 6 versions & 1 rubygems