Example #1
0
 void HorzRepsum::generate(CodeGenerator& g, const std::string& mem,
                           const std::vector<int>& arg, const std::vector<int>& res) const {
   int nnz = sparsity().nnz();
   g.body << "  " << g.fill(g.work(res[0], nnz), nnz, "0") << endl;
   g.body << "  for (i=0;i<" << n_ << ";++i) {" << endl;
   g.body << "    for (j=0;j<" << nnz << ";++j) {" << endl;
   g.body << "      " << g.work(res[0], nnz)<< "[j] += " <<
     g.work(arg[0], dep(0).nnz()) << "[j+i*" << nnz << "];" << endl;
   g.body << "    }" << endl;
   g.body << "  }" << endl;
 }