Created
November 10, 2017 02:37
-
-
Save eteq/6d2e253b510b8c6ee8583cea64d9261c to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| { | |
| "cells": [ | |
| { | |
| "cell_type": "markdown", | |
| "metadata": {}, | |
| "source": [ | |
| "# Easy(ish) notebook-based inner-loop optimization with C" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 1, | |
| "metadata": { | |
| "collapsed": true | |
| }, | |
| "outputs": [], | |
| "source": [ | |
| "import numpy as np" | |
| ] | |
| }, | |
| { | |
| "cell_type": "markdown", | |
| "metadata": {}, | |
| "source": [ | |
| "## Basic Python " | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 2, | |
| "metadata": { | |
| "collapsed": true | |
| }, | |
| "outputs": [], | |
| "source": [ | |
| "def py_addmul(inarr, tomul, toadd):\n", | |
| " out = [val*tomul + toadd for val in inarr]\n", | |
| " return np.array(out)\n", | |
| "\n", | |
| "def np_addmul(inarr, tomul, toadd):\n", | |
| " return inarr*tomul + toadd " | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 3, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "data": { | |
| "text/plain": [ | |
| "array([ 1.2, 3.2, 5.2, 7.2, 9.2, 11.2, 13.2, 15.2, 17.2, 19.2])" | |
| ] | |
| }, | |
| "execution_count": 3, | |
| "metadata": {}, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "py_addmul(np.arange(10, dtype=float), 2, 1.2)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 4, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "data": { | |
| "text/plain": [ | |
| "array([ 1.2, 3.2, 5.2, 7.2, 9.2, 11.2, 13.2, 15.2, 17.2, 19.2])" | |
| ] | |
| }, | |
| "execution_count": 4, | |
| "metadata": {}, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "np_addmul(np.arange(10, dtype=float), 2, 1.2)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "markdown", | |
| "metadata": {}, | |
| "source": [ | |
| "## Cython " | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 5, | |
| "metadata": { | |
| "collapsed": true | |
| }, | |
| "outputs": [], | |
| "source": [ | |
| "import Cython\n", | |
| "%load_ext Cython" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 6, | |
| "metadata": { | |
| "collapsed": true | |
| }, | |
| "outputs": [], | |
| "source": [ | |
| "%%cython\n", | |
| "\n", | |
| "import numpy as np\n", | |
| "\n", | |
| "def fast_addmul(inarr, tomul, toadd):\n", | |
| " outarr = np.empty_like(inarr)\n", | |
| " cdef double[:] inarr_memview = inarr\n", | |
| " \n", | |
| " c_fast_addmul(inarr_memview, outarr, len(inarr), toadd, tomul)\n", | |
| " \n", | |
| " return outarr\n", | |
| "\n", | |
| "cdef c_fast_addmul(double[:] inarr, double[:] outarr, int arrlen, double toadd, double tomul):\n", | |
| " cdef int i\n", | |
| " for i in range(arrlen):\n", | |
| " outarr[i] = inarr[i]*tomul + toadd" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 7, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "data": { | |
| "text/plain": [ | |
| "array([ 1.2, 3.2, 5.2, 7.2, 9.2, 11.2, 13.2, 15.2, 17.2, 19.2])" | |
| ] | |
| }, | |
| "execution_count": 7, | |
| "metadata": {}, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "fast_addmul(np.arange(10, dtype=float), 2, 1.2)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 8, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "name": "stdout", | |
| "output_type": "stream", | |
| "text": [ | |
| "41 µs ± 1.69 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)\n", | |
| "2.53 µs ± 94.2 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)\n", | |
| "2.08 µs ± 45.9 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)\n" | |
| ] | |
| } | |
| ], | |
| "source": [ | |
| "n = 100\n", | |
| "\n", | |
| "arr = np.arange(n, dtype=float)\n", | |
| "\n", | |
| "\n", | |
| "%timeit py_addmul(arr, 1.2, 2)\n", | |
| "%timeit np_addmul(arr, 1.2, 2)\n", | |
| "%timeit fast_addmul(arr, 1.2, 2)\n", | |
| "None" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 9, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "name": "stdout", | |
| "output_type": "stream", | |
| "text": [ | |
| "423 ms ± 29.7 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)\n", | |
| "3.15 ms ± 80.8 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)\n", | |
| "2.48 ms ± 40.9 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)\n" | |
| ] | |
| } | |
| ], | |
| "source": [ | |
| "n = 1000000\n", | |
| "\n", | |
| "arr = np.arange(n, dtype=float)\n", | |
| "\n", | |
| "%timeit py_addmul(arr, 1.2, 2)\n", | |
| "%timeit np_addmul(arr, 1.2, 2)\n", | |
| "%timeit fast_addmul(arr, 1.2, 2)\n", | |
| "None" | |
| ] | |
| }, | |
| { | |
| "cell_type": "markdown", | |
| "metadata": {}, | |
| "source": [ | |
| "### Optimize " | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 10, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "data": { | |
| "text/html": [ | |
| "<!DOCTYPE html>\n", | |
| "<!-- Generated by Cython 0.25.2 -->\n", | |
| "<html>\n", | |
| "<head>\n", | |
| " <meta http-equiv=\"Content-Type\" content=\"text/html; charset=utf-8\" />\n", | |
| " <title>Cython: _cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1.pyx</title>\n", | |
| " <style type=\"text/css\">\n", | |
| " \n", | |
| "body.cython { font-family: courier; font-size: 12; }\n", | |
| "\n", | |
| ".cython.tag { }\n", | |
| ".cython.line { margin: 0em }\n", | |
| ".cython.code { font-size: 9; color: #444444; display: none; margin: 0px 0px 0px 8px; border-left: 8px none; }\n", | |
| "\n", | |
| ".cython.line .run { background-color: #B0FFB0; }\n", | |
| ".cython.line .mis { background-color: #FFB0B0; }\n", | |
| ".cython.code.run { border-left: 8px solid #B0FFB0; }\n", | |
| ".cython.code.mis { border-left: 8px solid #FFB0B0; }\n", | |
| "\n", | |
| ".cython.code .py_c_api { color: red; }\n", | |
| ".cython.code .py_macro_api { color: #FF7000; }\n", | |
| ".cython.code .pyx_c_api { color: #FF3000; }\n", | |
| ".cython.code .pyx_macro_api { color: #FF7000; }\n", | |
| ".cython.code .refnanny { color: #FFA000; }\n", | |
| ".cython.code .trace { color: #FFA000; }\n", | |
| ".cython.code .error_goto { color: #FFA000; }\n", | |
| "\n", | |
| ".cython.code .coerce { color: #008000; border: 1px dotted #008000 }\n", | |
| ".cython.code .py_attr { color: #FF0000; font-weight: bold; }\n", | |
| ".cython.code .c_attr { color: #0000FF; }\n", | |
| ".cython.code .py_call { color: #FF0000; font-weight: bold; }\n", | |
| ".cython.code .c_call { color: #0000FF; }\n", | |
| "\n", | |
| ".cython.score-0 {background-color: #FFFFff;}\n", | |
| ".cython.score-1 {background-color: #FFFFe7;}\n", | |
| ".cython.score-2 {background-color: #FFFFd4;}\n", | |
| ".cython.score-3 {background-color: #FFFFc4;}\n", | |
| ".cython.score-4 {background-color: #FFFFb6;}\n", | |
| ".cython.score-5 {background-color: #FFFFaa;}\n", | |
| ".cython.score-6 {background-color: #FFFF9f;}\n", | |
| ".cython.score-7 {background-color: #FFFF96;}\n", | |
| ".cython.score-8 {background-color: #FFFF8d;}\n", | |
| ".cython.score-9 {background-color: #FFFF86;}\n", | |
| ".cython.score-10 {background-color: #FFFF7f;}\n", | |
| ".cython.score-11 {background-color: #FFFF79;}\n", | |
| ".cython.score-12 {background-color: #FFFF73;}\n", | |
| ".cython.score-13 {background-color: #FFFF6e;}\n", | |
| ".cython.score-14 {background-color: #FFFF6a;}\n", | |
| ".cython.score-15 {background-color: #FFFF66;}\n", | |
| ".cython.score-16 {background-color: #FFFF62;}\n", | |
| ".cython.score-17 {background-color: #FFFF5e;}\n", | |
| ".cython.score-18 {background-color: #FFFF5b;}\n", | |
| ".cython.score-19 {background-color: #FFFF57;}\n", | |
| ".cython.score-20 {background-color: #FFFF55;}\n", | |
| ".cython.score-21 {background-color: #FFFF52;}\n", | |
| ".cython.score-22 {background-color: #FFFF4f;}\n", | |
| ".cython.score-23 {background-color: #FFFF4d;}\n", | |
| ".cython.score-24 {background-color: #FFFF4b;}\n", | |
| ".cython.score-25 {background-color: #FFFF48;}\n", | |
| ".cython.score-26 {background-color: #FFFF46;}\n", | |
| ".cython.score-27 {background-color: #FFFF44;}\n", | |
| ".cython.score-28 {background-color: #FFFF43;}\n", | |
| ".cython.score-29 {background-color: #FFFF41;}\n", | |
| ".cython.score-30 {background-color: #FFFF3f;}\n", | |
| ".cython.score-31 {background-color: #FFFF3e;}\n", | |
| ".cython.score-32 {background-color: #FFFF3c;}\n", | |
| ".cython.score-33 {background-color: #FFFF3b;}\n", | |
| ".cython.score-34 {background-color: #FFFF39;}\n", | |
| ".cython.score-35 {background-color: #FFFF38;}\n", | |
| ".cython.score-36 {background-color: #FFFF37;}\n", | |
| ".cython.score-37 {background-color: #FFFF36;}\n", | |
| ".cython.score-38 {background-color: #FFFF35;}\n", | |
| ".cython.score-39 {background-color: #FFFF34;}\n", | |
| ".cython.score-40 {background-color: #FFFF33;}\n", | |
| ".cython.score-41 {background-color: #FFFF32;}\n", | |
| ".cython.score-42 {background-color: #FFFF31;}\n", | |
| ".cython.score-43 {background-color: #FFFF30;}\n", | |
| ".cython.score-44 {background-color: #FFFF2f;}\n", | |
| ".cython.score-45 {background-color: #FFFF2e;}\n", | |
| ".cython.score-46 {background-color: #FFFF2d;}\n", | |
| ".cython.score-47 {background-color: #FFFF2c;}\n", | |
| ".cython.score-48 {background-color: #FFFF2b;}\n", | |
| ".cython.score-49 {background-color: #FFFF2b;}\n", | |
| ".cython.score-50 {background-color: #FFFF2a;}\n", | |
| ".cython.score-51 {background-color: #FFFF29;}\n", | |
| ".cython.score-52 {background-color: #FFFF29;}\n", | |
| ".cython.score-53 {background-color: #FFFF28;}\n", | |
| ".cython.score-54 {background-color: #FFFF27;}\n", | |
| ".cython.score-55 {background-color: #FFFF27;}\n", | |
| ".cython.score-56 {background-color: #FFFF26;}\n", | |
| ".cython.score-57 {background-color: #FFFF26;}\n", | |
| ".cython.score-58 {background-color: #FFFF25;}\n", | |
| ".cython.score-59 {background-color: #FFFF24;}\n", | |
| ".cython.score-60 {background-color: #FFFF24;}\n", | |
| ".cython.score-61 {background-color: #FFFF23;}\n", | |
| ".cython.score-62 {background-color: #FFFF23;}\n", | |
| ".cython.score-63 {background-color: #FFFF22;}\n", | |
| ".cython.score-64 {background-color: #FFFF22;}\n", | |
| ".cython.score-65 {background-color: #FFFF22;}\n", | |
| ".cython.score-66 {background-color: #FFFF21;}\n", | |
| ".cython.score-67 {background-color: #FFFF21;}\n", | |
| ".cython.score-68 {background-color: #FFFF20;}\n", | |
| ".cython.score-69 {background-color: #FFFF20;}\n", | |
| ".cython.score-70 {background-color: #FFFF1f;}\n", | |
| ".cython.score-71 {background-color: #FFFF1f;}\n", | |
| ".cython.score-72 {background-color: #FFFF1f;}\n", | |
| ".cython.score-73 {background-color: #FFFF1e;}\n", | |
| ".cython.score-74 {background-color: #FFFF1e;}\n", | |
| ".cython.score-75 {background-color: #FFFF1e;}\n", | |
| ".cython.score-76 {background-color: #FFFF1d;}\n", | |
| ".cython.score-77 {background-color: #FFFF1d;}\n", | |
| ".cython.score-78 {background-color: #FFFF1c;}\n", | |
| ".cython.score-79 {background-color: #FFFF1c;}\n", | |
| ".cython.score-80 {background-color: #FFFF1c;}\n", | |
| ".cython.score-81 {background-color: #FFFF1c;}\n", | |
| ".cython.score-82 {background-color: #FFFF1b;}\n", | |
| ".cython.score-83 {background-color: #FFFF1b;}\n", | |
| ".cython.score-84 {background-color: #FFFF1b;}\n", | |
| ".cython.score-85 {background-color: #FFFF1a;}\n", | |
| ".cython.score-86 {background-color: #FFFF1a;}\n", | |
| ".cython.score-87 {background-color: #FFFF1a;}\n", | |
| ".cython.score-88 {background-color: #FFFF1a;}\n", | |
| ".cython.score-89 {background-color: #FFFF19;}\n", | |
| ".cython.score-90 {background-color: #FFFF19;}\n", | |
| ".cython.score-91 {background-color: #FFFF19;}\n", | |
| ".cython.score-92 {background-color: #FFFF19;}\n", | |
| ".cython.score-93 {background-color: #FFFF18;}\n", | |
| ".cython.score-94 {background-color: #FFFF18;}\n", | |
| ".cython.score-95 {background-color: #FFFF18;}\n", | |
| ".cython.score-96 {background-color: #FFFF18;}\n", | |
| ".cython.score-97 {background-color: #FFFF17;}\n", | |
| ".cython.score-98 {background-color: #FFFF17;}\n", | |
| ".cython.score-99 {background-color: #FFFF17;}\n", | |
| ".cython.score-100 {background-color: #FFFF17;}\n", | |
| ".cython.score-101 {background-color: #FFFF16;}\n", | |
| ".cython.score-102 {background-color: #FFFF16;}\n", | |
| ".cython.score-103 {background-color: #FFFF16;}\n", | |
| ".cython.score-104 {background-color: #FFFF16;}\n", | |
| ".cython.score-105 {background-color: #FFFF16;}\n", | |
| ".cython.score-106 {background-color: #FFFF15;}\n", | |
| ".cython.score-107 {background-color: #FFFF15;}\n", | |
| ".cython.score-108 {background-color: #FFFF15;}\n", | |
| ".cython.score-109 {background-color: #FFFF15;}\n", | |
| ".cython.score-110 {background-color: #FFFF15;}\n", | |
| ".cython.score-111 {background-color: #FFFF15;}\n", | |
| ".cython.score-112 {background-color: #FFFF14;}\n", | |
| ".cython.score-113 {background-color: #FFFF14;}\n", | |
| ".cython.score-114 {background-color: #FFFF14;}\n", | |
| ".cython.score-115 {background-color: #FFFF14;}\n", | |
| ".cython.score-116 {background-color: #FFFF14;}\n", | |
| ".cython.score-117 {background-color: #FFFF14;}\n", | |
| ".cython.score-118 {background-color: #FFFF13;}\n", | |
| ".cython.score-119 {background-color: #FFFF13;}\n", | |
| ".cython.score-120 {background-color: #FFFF13;}\n", | |
| ".cython.score-121 {background-color: #FFFF13;}\n", | |
| ".cython.score-122 {background-color: #FFFF13;}\n", | |
| ".cython.score-123 {background-color: #FFFF13;}\n", | |
| ".cython.score-124 {background-color: #FFFF13;}\n", | |
| ".cython.score-125 {background-color: #FFFF12;}\n", | |
| ".cython.score-126 {background-color: #FFFF12;}\n", | |
| ".cython.score-127 {background-color: #FFFF12;}\n", | |
| ".cython.score-128 {background-color: #FFFF12;}\n", | |
| ".cython.score-129 {background-color: #FFFF12;}\n", | |
| ".cython.score-130 {background-color: #FFFF12;}\n", | |
| ".cython.score-131 {background-color: #FFFF12;}\n", | |
| ".cython.score-132 {background-color: #FFFF11;}\n", | |
| ".cython.score-133 {background-color: #FFFF11;}\n", | |
| ".cython.score-134 {background-color: #FFFF11;}\n", | |
| ".cython.score-135 {background-color: #FFFF11;}\n", | |
| ".cython.score-136 {background-color: #FFFF11;}\n", | |
| ".cython.score-137 {background-color: #FFFF11;}\n", | |
| ".cython.score-138 {background-color: #FFFF11;}\n", | |
| ".cython.score-139 {background-color: #FFFF11;}\n", | |
| ".cython.score-140 {background-color: #FFFF11;}\n", | |
| ".cython.score-141 {background-color: #FFFF10;}\n", | |
| ".cython.score-142 {background-color: #FFFF10;}\n", | |
| ".cython.score-143 {background-color: #FFFF10;}\n", | |
| ".cython.score-144 {background-color: #FFFF10;}\n", | |
| ".cython.score-145 {background-color: #FFFF10;}\n", | |
| ".cython.score-146 {background-color: #FFFF10;}\n", | |
| ".cython.score-147 {background-color: #FFFF10;}\n", | |
| ".cython.score-148 {background-color: #FFFF10;}\n", | |
| ".cython.score-149 {background-color: #FFFF10;}\n", | |
| ".cython.score-150 {background-color: #FFFF0f;}\n", | |
| ".cython.score-151 {background-color: #FFFF0f;}\n", | |
| ".cython.score-152 {background-color: #FFFF0f;}\n", | |
| ".cython.score-153 {background-color: #FFFF0f;}\n", | |
| ".cython.score-154 {background-color: #FFFF0f;}\n", | |
| ".cython.score-155 {background-color: #FFFF0f;}\n", | |
| ".cython.score-156 {background-color: #FFFF0f;}\n", | |
| ".cython.score-157 {background-color: #FFFF0f;}\n", | |
| ".cython.score-158 {background-color: #FFFF0f;}\n", | |
| ".cython.score-159 {background-color: #FFFF0f;}\n", | |
| ".cython.score-160 {background-color: #FFFF0f;}\n", | |
| ".cython.score-161 {background-color: #FFFF0e;}\n", | |
| ".cython.score-162 {background-color: #FFFF0e;}\n", | |
| ".cython.score-163 {background-color: #FFFF0e;}\n", | |
| ".cython.score-164 {background-color: #FFFF0e;}\n", | |
| ".cython.score-165 {background-color: #FFFF0e;}\n", | |
| ".cython.score-166 {background-color: #FFFF0e;}\n", | |
| ".cython.score-167 {background-color: #FFFF0e;}\n", | |
| ".cython.score-168 {background-color: #FFFF0e;}\n", | |
| ".cython.score-169 {background-color: #FFFF0e;}\n", | |
| ".cython.score-170 {background-color: #FFFF0e;}\n", | |
| ".cython.score-171 {background-color: #FFFF0e;}\n", | |
| ".cython.score-172 {background-color: #FFFF0e;}\n", | |
| ".cython.score-173 {background-color: #FFFF0d;}\n", | |
| ".cython.score-174 {background-color: #FFFF0d;}\n", | |
| ".cython.score-175 {background-color: #FFFF0d;}\n", | |
| ".cython.score-176 {background-color: #FFFF0d;}\n", | |
| ".cython.score-177 {background-color: #FFFF0d;}\n", | |
| ".cython.score-178 {background-color: #FFFF0d;}\n", | |
| ".cython.score-179 {background-color: #FFFF0d;}\n", | |
| ".cython.score-180 {background-color: #FFFF0d;}\n", | |
| ".cython.score-181 {background-color: #FFFF0d;}\n", | |
| ".cython.score-182 {background-color: #FFFF0d;}\n", | |
| ".cython.score-183 {background-color: #FFFF0d;}\n", | |
| ".cython.score-184 {background-color: #FFFF0d;}\n", | |
| ".cython.score-185 {background-color: #FFFF0d;}\n", | |
| ".cython.score-186 {background-color: #FFFF0d;}\n", | |
| ".cython.score-187 {background-color: #FFFF0c;}\n", | |
| ".cython.score-188 {background-color: #FFFF0c;}\n", | |
| ".cython.score-189 {background-color: #FFFF0c;}\n", | |
| ".cython.score-190 {background-color: #FFFF0c;}\n", | |
| ".cython.score-191 {background-color: #FFFF0c;}\n", | |
| ".cython.score-192 {background-color: #FFFF0c;}\n", | |
| ".cython.score-193 {background-color: #FFFF0c;}\n", | |
| ".cython.score-194 {background-color: #FFFF0c;}\n", | |
| ".cython.score-195 {background-color: #FFFF0c;}\n", | |
| ".cython.score-196 {background-color: #FFFF0c;}\n", | |
| ".cython.score-197 {background-color: #FFFF0c;}\n", | |
| ".cython.score-198 {background-color: #FFFF0c;}\n", | |
| ".cython.score-199 {background-color: #FFFF0c;}\n", | |
| ".cython.score-200 {background-color: #FFFF0c;}\n", | |
| ".cython.score-201 {background-color: #FFFF0c;}\n", | |
| ".cython.score-202 {background-color: #FFFF0c;}\n", | |
| ".cython.score-203 {background-color: #FFFF0b;}\n", | |
| ".cython.score-204 {background-color: #FFFF0b;}\n", | |
| ".cython.score-205 {background-color: #FFFF0b;}\n", | |
| ".cython.score-206 {background-color: #FFFF0b;}\n", | |
| ".cython.score-207 {background-color: #FFFF0b;}\n", | |
| ".cython.score-208 {background-color: #FFFF0b;}\n", | |
| ".cython.score-209 {background-color: #FFFF0b;}\n", | |
| ".cython.score-210 {background-color: #FFFF0b;}\n", | |
| ".cython.score-211 {background-color: #FFFF0b;}\n", | |
| ".cython.score-212 {background-color: #FFFF0b;}\n", | |
| ".cython.score-213 {background-color: #FFFF0b;}\n", | |
| ".cython.score-214 {background-color: #FFFF0b;}\n", | |
| ".cython.score-215 {background-color: #FFFF0b;}\n", | |
| ".cython.score-216 {background-color: #FFFF0b;}\n", | |
| ".cython.score-217 {background-color: #FFFF0b;}\n", | |
| ".cython.score-218 {background-color: #FFFF0b;}\n", | |
| ".cython.score-219 {background-color: #FFFF0b;}\n", | |
| ".cython.score-220 {background-color: #FFFF0b;}\n", | |
| ".cython.score-221 {background-color: #FFFF0b;}\n", | |
| ".cython.score-222 {background-color: #FFFF0a;}\n", | |
| ".cython.score-223 {background-color: #FFFF0a;}\n", | |
| ".cython.score-224 {background-color: #FFFF0a;}\n", | |
| ".cython.score-225 {background-color: #FFFF0a;}\n", | |
| ".cython.score-226 {background-color: #FFFF0a;}\n", | |
| ".cython.score-227 {background-color: #FFFF0a;}\n", | |
| ".cython.score-228 {background-color: #FFFF0a;}\n", | |
| ".cython.score-229 {background-color: #FFFF0a;}\n", | |
| ".cython.score-230 {background-color: #FFFF0a;}\n", | |
| ".cython.score-231 {background-color: #FFFF0a;}\n", | |
| ".cython.score-232 {background-color: #FFFF0a;}\n", | |
| ".cython.score-233 {background-color: #FFFF0a;}\n", | |
| ".cython.score-234 {background-color: #FFFF0a;}\n", | |
| ".cython.score-235 {background-color: #FFFF0a;}\n", | |
| ".cython.score-236 {background-color: #FFFF0a;}\n", | |
| ".cython.score-237 {background-color: #FFFF0a;}\n", | |
| ".cython.score-238 {background-color: #FFFF0a;}\n", | |
| ".cython.score-239 {background-color: #FFFF0a;}\n", | |
| ".cython.score-240 {background-color: #FFFF0a;}\n", | |
| ".cython.score-241 {background-color: #FFFF0a;}\n", | |
| ".cython.score-242 {background-color: #FFFF0a;}\n", | |
| ".cython.score-243 {background-color: #FFFF0a;}\n", | |
| ".cython.score-244 {background-color: #FFFF0a;}\n", | |
| ".cython.score-245 {background-color: #FFFF0a;}\n", | |
| ".cython.score-246 {background-color: #FFFF09;}\n", | |
| ".cython.score-247 {background-color: #FFFF09;}\n", | |
| ".cython.score-248 {background-color: #FFFF09;}\n", | |
| ".cython.score-249 {background-color: #FFFF09;}\n", | |
| ".cython.score-250 {background-color: #FFFF09;}\n", | |
| ".cython.score-251 {background-color: #FFFF09;}\n", | |
| ".cython.score-252 {background-color: #FFFF09;}\n", | |
| ".cython.score-253 {background-color: #FFFF09;}\n", | |
| ".cython.score-254 {background-color: #FFFF09;}\n", | |
| ".cython .hll { background-color: #ffffcc }\n", | |
| ".cython { background: #f8f8f8; }\n", | |
| ".cython .c { color: #408080; font-style: italic } /* Comment */\n", | |
| ".cython .err { border: 1px solid #FF0000 } /* Error */\n", | |
| ".cython .k { color: #008000; font-weight: bold } /* Keyword */\n", | |
| ".cython .o { color: #666666 } /* Operator */\n", | |
| ".cython .ch { color: #408080; font-style: italic } /* Comment.Hashbang */\n", | |
| ".cython .cm { color: #408080; font-style: italic } /* Comment.Multiline */\n", | |
| ".cython .cp { color: #BC7A00 } /* Comment.Preproc */\n", | |
| ".cython .cpf { color: #408080; font-style: italic } /* Comment.PreprocFile */\n", | |
| ".cython .c1 { color: #408080; font-style: italic } /* Comment.Single */\n", | |
| ".cython .cs { color: #408080; font-style: italic } /* Comment.Special */\n", | |
| ".cython .gd { color: #A00000 } /* Generic.Deleted */\n", | |
| ".cython .ge { font-style: italic } /* Generic.Emph */\n", | |
| ".cython .gr { color: #FF0000 } /* Generic.Error */\n", | |
| ".cython .gh { color: #000080; font-weight: bold } /* Generic.Heading */\n", | |
| ".cython .gi { color: #00A000 } /* Generic.Inserted */\n", | |
| ".cython .go { color: #888888 } /* Generic.Output */\n", | |
| ".cython .gp { color: #000080; font-weight: bold } /* Generic.Prompt */\n", | |
| ".cython .gs { font-weight: bold } /* Generic.Strong */\n", | |
| ".cython .gu { color: #800080; font-weight: bold } /* Generic.Subheading */\n", | |
| ".cython .gt { color: #0044DD } /* Generic.Traceback */\n", | |
| ".cython .kc { color: #008000; font-weight: bold } /* Keyword.Constant */\n", | |
| ".cython .kd { color: #008000; font-weight: bold } /* Keyword.Declaration */\n", | |
| ".cython .kn { color: #008000; font-weight: bold } /* Keyword.Namespace */\n", | |
| ".cython .kp { color: #008000 } /* Keyword.Pseudo */\n", | |
| ".cython .kr { color: #008000; font-weight: bold } /* Keyword.Reserved */\n", | |
| ".cython .kt { color: #B00040 } /* Keyword.Type */\n", | |
| ".cython .m { color: #666666 } /* Literal.Number */\n", | |
| ".cython .s { color: #BA2121 } /* Literal.String */\n", | |
| ".cython .na { color: #7D9029 } /* Name.Attribute */\n", | |
| ".cython .nb { color: #008000 } /* Name.Builtin */\n", | |
| ".cython .nc { color: #0000FF; font-weight: bold } /* Name.Class */\n", | |
| ".cython .no { color: #880000 } /* Name.Constant */\n", | |
| ".cython .nd { color: #AA22FF } /* Name.Decorator */\n", | |
| ".cython .ni { color: #999999; font-weight: bold } /* Name.Entity */\n", | |
| ".cython .ne { color: #D2413A; font-weight: bold } /* Name.Exception */\n", | |
| ".cython .nf { color: #0000FF } /* Name.Function */\n", | |
| ".cython .nl { color: #A0A000 } /* Name.Label */\n", | |
| ".cython .nn { color: #0000FF; font-weight: bold } /* Name.Namespace */\n", | |
| ".cython .nt { color: #008000; font-weight: bold } /* Name.Tag */\n", | |
| ".cython .nv { color: #19177C } /* Name.Variable */\n", | |
| ".cython .ow { color: #AA22FF; font-weight: bold } /* Operator.Word */\n", | |
| ".cython .w { color: #bbbbbb } /* Text.Whitespace */\n", | |
| ".cython .mb { color: #666666 } /* Literal.Number.Bin */\n", | |
| ".cython .mf { color: #666666 } /* Literal.Number.Float */\n", | |
| ".cython .mh { color: #666666 } /* Literal.Number.Hex */\n", | |
| ".cython .mi { color: #666666 } /* Literal.Number.Integer */\n", | |
| ".cython .mo { color: #666666 } /* Literal.Number.Oct */\n", | |
| ".cython .sb { color: #BA2121 } /* Literal.String.Backtick */\n", | |
| ".cython .sc { color: #BA2121 } /* Literal.String.Char */\n", | |
| ".cython .sd { color: #BA2121; font-style: italic } /* Literal.String.Doc */\n", | |
| ".cython .s2 { color: #BA2121 } /* Literal.String.Double */\n", | |
| ".cython .se { color: #BB6622; font-weight: bold } /* Literal.String.Escape */\n", | |
| ".cython .sh { color: #BA2121 } /* Literal.String.Heredoc */\n", | |
| ".cython .si { color: #BB6688; font-weight: bold } /* Literal.String.Interpol */\n", | |
| ".cython .sx { color: #008000 } /* Literal.String.Other */\n", | |
| ".cython .sr { color: #BB6688 } /* Literal.String.Regex */\n", | |
| ".cython .s1 { color: #BA2121 } /* Literal.String.Single */\n", | |
| ".cython .ss { color: #19177C } /* Literal.String.Symbol */\n", | |
| ".cython .bp { color: #008000 } /* Name.Builtin.Pseudo */\n", | |
| ".cython .vc { color: #19177C } /* Name.Variable.Class */\n", | |
| ".cython .vg { color: #19177C } /* Name.Variable.Global */\n", | |
| ".cython .vi { color: #19177C } /* Name.Variable.Instance */\n", | |
| ".cython .il { color: #666666 } /* Literal.Number.Integer.Long */\n", | |
| " </style>\n", | |
| " <script>\n", | |
| " function toggleDiv(id) {\n", | |
| " theDiv = id.nextElementSibling\n", | |
| " if (theDiv.style.display != 'block') theDiv.style.display = 'block';\n", | |
| " else theDiv.style.display = 'none';\n", | |
| " }\n", | |
| " </script>\n", | |
| "</head>\n", | |
| "<body class=\"cython\">\n", | |
| "<p><span style=\"border-bottom: solid 1px grey;\">Generated by Cython 0.25.2</span></p>\n", | |
| "<p>\n", | |
| " <span style=\"background-color: #FFFF00\">Yellow lines</span> hint at Python interaction.<br />\n", | |
| " Click on a line that starts with a \"<code>+</code>\" to see the C code that Cython generated for it.\n", | |
| "</p>\n", | |
| "<div class=\"cython\"><pre class=\"cython line score-0\"> <span class=\"\">01</span>: </pre>\n", | |
| "<pre class=\"cython line score-19\" onclick='toggleDiv(this)'>+<span class=\"\">02</span>: <span class=\"k\">import</span> <span class=\"nn\">numpy</span> <span class=\"k\">as</span> <span class=\"nn\">np</span></pre>\n", | |
| "<pre class='cython code score-19 '> __pyx_t_1 = <span class='pyx_c_api'>__Pyx_Import</span>(__pyx_n_s_numpy, 0, 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 2, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " if (<span class='py_c_api'>PyDict_SetItem</span>(__pyx_d, __pyx_n_s_np, __pyx_t_1) < 0) __PYX_ERR(0, 2, __pyx_L1_error)\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_1); __pyx_t_1 = 0;\n", | |
| "/* … */\n", | |
| " __pyx_t_1 = <span class='py_c_api'>PyDict_New</span>(); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 2, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " if (<span class='py_c_api'>PyDict_SetItem</span>(__pyx_d, __pyx_n_s_test, __pyx_t_1) < 0) __PYX_ERR(0, 2, __pyx_L1_error)\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_1); __pyx_t_1 = 0;\n", | |
| "</pre><pre class=\"cython line score-0\"> <span class=\"\">03</span>: </pre>\n", | |
| "<pre class=\"cython line score-59\" onclick='toggleDiv(this)'>+<span class=\"\">04</span>: <span class=\"k\">def</span> <span class=\"nf\">fast_addmul</span><span class=\"p\">(</span><span class=\"n\">inarr</span><span class=\"p\">,</span> <span class=\"n\">tomul</span><span class=\"p\">,</span> <span class=\"n\">toadd</span><span class=\"p\">):</span></pre>\n", | |
| "<pre class='cython code score-59 '>/* Python wrapper */\n", | |
| "static PyObject *__pyx_pw_46_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1_1fast_addmul(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/\n", | |
| "static PyMethodDef __pyx_mdef_46_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1_1fast_addmul = {\"fast_addmul\", (PyCFunction)__pyx_pw_46_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1_1fast_addmul, METH_VARARGS|METH_KEYWORDS, 0};\n", | |
| "static PyObject *__pyx_pw_46_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1_1fast_addmul(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {\n", | |
| " PyObject *__pyx_v_inarr = 0;\n", | |
| " PyObject *__pyx_v_tomul = 0;\n", | |
| " PyObject *__pyx_v_toadd = 0;\n", | |
| " PyObject *__pyx_r = 0;\n", | |
| " <span class='refnanny'>__Pyx_RefNannyDeclarations</span>\n", | |
| " <span class='refnanny'>__Pyx_RefNannySetupContext</span>(\"fast_addmul (wrapper)\", 0);\n", | |
| " {\n", | |
| " static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_inarr,&__pyx_n_s_tomul,&__pyx_n_s_toadd,0};\n", | |
| " PyObject* values[3] = {0,0,0};\n", | |
| " if (unlikely(__pyx_kwds)) {\n", | |
| " Py_ssize_t kw_args;\n", | |
| " const Py_ssize_t pos_args = <span class='py_macro_api'>PyTuple_GET_SIZE</span>(__pyx_args);\n", | |
| " switch (pos_args) {\n", | |
| " case 3: values[2] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 2);\n", | |
| " case 2: values[1] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 1);\n", | |
| " case 1: values[0] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 0);\n", | |
| " case 0: break;\n", | |
| " default: goto __pyx_L5_argtuple_error;\n", | |
| " }\n", | |
| " kw_args = <span class='py_c_api'>PyDict_Size</span>(__pyx_kwds);\n", | |
| " switch (pos_args) {\n", | |
| " case 0:\n", | |
| " if (likely((values[0] = <span class='py_c_api'>PyDict_GetItem</span>(__pyx_kwds, __pyx_n_s_inarr)) != 0)) kw_args--;\n", | |
| " else goto __pyx_L5_argtuple_error;\n", | |
| " case 1:\n", | |
| " if (likely((values[1] = <span class='py_c_api'>PyDict_GetItem</span>(__pyx_kwds, __pyx_n_s_tomul)) != 0)) kw_args--;\n", | |
| " else {\n", | |
| " <span class='pyx_c_api'>__Pyx_RaiseArgtupleInvalid</span>(\"fast_addmul\", 1, 3, 3, 1); __PYX_ERR(0, 4, __pyx_L3_error)\n", | |
| " }\n", | |
| " case 2:\n", | |
| " if (likely((values[2] = <span class='py_c_api'>PyDict_GetItem</span>(__pyx_kwds, __pyx_n_s_toadd)) != 0)) kw_args--;\n", | |
| " else {\n", | |
| " <span class='pyx_c_api'>__Pyx_RaiseArgtupleInvalid</span>(\"fast_addmul\", 1, 3, 3, 2); __PYX_ERR(0, 4, __pyx_L3_error)\n", | |
| " }\n", | |
| " }\n", | |
| " if (unlikely(kw_args > 0)) {\n", | |
| " if (unlikely(<span class='pyx_c_api'>__Pyx_ParseOptionalKeywords</span>(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, \"fast_addmul\") < 0)) __PYX_ERR(0, 4, __pyx_L3_error)\n", | |
| " }\n", | |
| " } else if (<span class='py_macro_api'>PyTuple_GET_SIZE</span>(__pyx_args) != 3) {\n", | |
| " goto __pyx_L5_argtuple_error;\n", | |
| " } else {\n", | |
| " values[0] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 0);\n", | |
| " values[1] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 1);\n", | |
| " values[2] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 2);\n", | |
| " }\n", | |
| " __pyx_v_inarr = values[0];\n", | |
| " __pyx_v_tomul = values[1];\n", | |
| " __pyx_v_toadd = values[2];\n", | |
| " }\n", | |
| " goto __pyx_L4_argument_unpacking_done;\n", | |
| " __pyx_L5_argtuple_error:;\n", | |
| " <span class='pyx_c_api'>__Pyx_RaiseArgtupleInvalid</span>(\"fast_addmul\", 1, 3, 3, <span class='py_macro_api'>PyTuple_GET_SIZE</span>(__pyx_args)); __PYX_ERR(0, 4, __pyx_L3_error)\n", | |
| " __pyx_L3_error:;\n", | |
| " <span class='pyx_c_api'>__Pyx_AddTraceback</span>(\"_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1.fast_addmul\", __pyx_clineno, __pyx_lineno, __pyx_filename);\n", | |
| " <span class='refnanny'>__Pyx_RefNannyFinishContext</span>();\n", | |
| " return NULL;\n", | |
| " __pyx_L4_argument_unpacking_done:;\n", | |
| " __pyx_r = __pyx_pf_46_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1_fast_addmul(__pyx_self, __pyx_v_inarr, __pyx_v_tomul, __pyx_v_toadd);\n", | |
| "\n", | |
| " /* function exit code */\n", | |
| " <span class='refnanny'>__Pyx_RefNannyFinishContext</span>();\n", | |
| " return __pyx_r;\n", | |
| "}\n", | |
| "\n", | |
| "static PyObject *__pyx_pf_46_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1_fast_addmul(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_inarr, PyObject *__pyx_v_tomul, PyObject *__pyx_v_toadd) {\n", | |
| " PyObject *__pyx_v_outarr = NULL;\n", | |
| " __Pyx_memviewslice __pyx_v_inarr_memview = { 0, 0, { 0 }, { 0 }, { 0 } };\n", | |
| " PyObject *__pyx_r = NULL;\n", | |
| " <span class='refnanny'>__Pyx_RefNannyDeclarations</span>\n", | |
| " <span class='refnanny'>__Pyx_RefNannySetupContext</span>(\"fast_addmul\", 0);\n", | |
| "/* … */\n", | |
| " /* function exit code */\n", | |
| " __pyx_L1_error:;\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_1);\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_2);\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_3);\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_4);\n", | |
| " __PYX_XDEC_MEMVIEW(&__pyx_t_5, 1);\n", | |
| " <span class='pyx_c_api'>__Pyx_AddTraceback</span>(\"_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1.fast_addmul\", __pyx_clineno, __pyx_lineno, __pyx_filename);\n", | |
| " __pyx_r = NULL;\n", | |
| " __pyx_L0:;\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_v_outarr);\n", | |
| " __PYX_XDEC_MEMVIEW(&__pyx_v_inarr_memview, 1);\n", | |
| " <span class='refnanny'>__Pyx_XGIVEREF</span>(__pyx_r);\n", | |
| " <span class='refnanny'>__Pyx_RefNannyFinishContext</span>();\n", | |
| " return __pyx_r;\n", | |
| "}\n", | |
| "/* … */\n", | |
| " __pyx_tuple__14 = <span class='py_c_api'>PyTuple_Pack</span>(5, __pyx_n_s_inarr, __pyx_n_s_tomul, __pyx_n_s_toadd, __pyx_n_s_outarr, __pyx_n_s_inarr_memview); if (unlikely(!__pyx_tuple__14)) __PYX_ERR(0, 4, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_tuple__14);\n", | |
| " <span class='refnanny'>__Pyx_GIVEREF</span>(__pyx_tuple__14);\n", | |
| "/* … */\n", | |
| " __pyx_t_1 = PyCFunction_NewEx(&__pyx_mdef_46_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1_1fast_addmul, NULL, __pyx_n_s_cython_magic_d6ed5f1fc7de3b51bb); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 4, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " if (<span class='py_c_api'>PyDict_SetItem</span>(__pyx_d, __pyx_n_s_fast_addmul, __pyx_t_1) < 0) __PYX_ERR(0, 4, __pyx_L1_error)\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_1); __pyx_t_1 = 0;\n", | |
| " __pyx_codeobj__15 = (PyObject*)<span class='pyx_c_api'>__Pyx_PyCode_New</span>(3, 0, 5, 0, 0, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__14, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_Users_erik_ipython_cython__cyth, __pyx_n_s_fast_addmul, 4, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__15)) __PYX_ERR(0, 4, __pyx_L1_error)\n", | |
| "</pre><pre class=\"cython line score-42\" onclick='toggleDiv(this)'>+<span class=\"\">05</span>: <span class=\"n\">outarr</span> <span class=\"o\">=</span> <span class=\"n\">np</span><span class=\"o\">.</span><span class=\"n\">empty_like</span><span class=\"p\">(</span><span class=\"n\">inarr</span><span class=\"p\">)</span></pre>\n", | |
| "<pre class='cython code score-42 '> __pyx_t_2 = <span class='pyx_c_api'>__Pyx_GetModuleGlobalName</span>(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_2);\n", | |
| " __pyx_t_3 = <span class='pyx_c_api'>__Pyx_PyObject_GetAttrStr</span>(__pyx_t_2, __pyx_n_s_empty_like); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_3);\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_2); __pyx_t_2 = 0;\n", | |
| " __pyx_t_2 = NULL;\n", | |
| " if (CYTHON_UNPACK_METHODS && unlikely(<span class='py_c_api'>PyMethod_Check</span>(__pyx_t_3))) {\n", | |
| " __pyx_t_2 = <span class='py_macro_api'>PyMethod_GET_SELF</span>(__pyx_t_3);\n", | |
| " if (likely(__pyx_t_2)) {\n", | |
| " PyObject* function = <span class='py_macro_api'>PyMethod_GET_FUNCTION</span>(__pyx_t_3);\n", | |
| " <span class='pyx_macro_api'>__Pyx_INCREF</span>(__pyx_t_2);\n", | |
| " <span class='pyx_macro_api'>__Pyx_INCREF</span>(function);\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF_SET</span>(__pyx_t_3, function);\n", | |
| " }\n", | |
| " }\n", | |
| " if (!__pyx_t_2) {\n", | |
| " __pyx_t_1 = <span class='pyx_c_api'>__Pyx_PyObject_CallOneArg</span>(__pyx_t_3, __pyx_v_inarr); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " } else {\n", | |
| " #if CYTHON_FAST_PYCALL\n", | |
| " if (<span class='py_c_api'>PyFunction_Check</span>(__pyx_t_3)) {\n", | |
| " PyObject *__pyx_temp[2] = {__pyx_t_2, __pyx_v_inarr};\n", | |
| " __pyx_t_1 = <span class='pyx_c_api'>__Pyx_PyFunction_FastCall</span>(__pyx_t_3, __pyx_temp+1-1, 1+1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_2); __pyx_t_2 = 0;\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " } else\n", | |
| " #endif\n", | |
| " #if CYTHON_FAST_PYCCALL\n", | |
| " if (<span class='pyx_c_api'>__Pyx_PyFastCFunction_Check</span>(__pyx_t_3)) {\n", | |
| " PyObject *__pyx_temp[2] = {__pyx_t_2, __pyx_v_inarr};\n", | |
| " __pyx_t_1 = <span class='pyx_c_api'>__Pyx_PyCFunction_FastCall</span>(__pyx_t_3, __pyx_temp+1-1, 1+1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_2); __pyx_t_2 = 0;\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " } else\n", | |
| " #endif\n", | |
| " {\n", | |
| " __pyx_t_4 = <span class='py_c_api'>PyTuple_New</span>(1+1); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_4);\n", | |
| " <span class='refnanny'>__Pyx_GIVEREF</span>(__pyx_t_2); <span class='py_macro_api'>PyTuple_SET_ITEM</span>(__pyx_t_4, 0, __pyx_t_2); __pyx_t_2 = NULL;\n", | |
| " <span class='pyx_macro_api'>__Pyx_INCREF</span>(__pyx_v_inarr);\n", | |
| " <span class='refnanny'>__Pyx_GIVEREF</span>(__pyx_v_inarr);\n", | |
| " <span class='py_macro_api'>PyTuple_SET_ITEM</span>(__pyx_t_4, 0+1, __pyx_v_inarr);\n", | |
| " __pyx_t_1 = <span class='pyx_c_api'>__Pyx_PyObject_Call</span>(__pyx_t_3, __pyx_t_4, NULL); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_4); __pyx_t_4 = 0;\n", | |
| " }\n", | |
| " }\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_3); __pyx_t_3 = 0;\n", | |
| " __pyx_v_outarr = __pyx_t_1;\n", | |
| " __pyx_t_1 = 0;\n", | |
| "</pre><pre class=\"cython line score-2\" onclick='toggleDiv(this)'>+<span class=\"\">06</span>: <span class=\"k\">cdef</span> <span class=\"kt\">double</span>[<span class=\"p\">:]</span> <span class=\"n\">inarr_memview</span> <span class=\"o\">=</span> <span class=\"n\">inarr</span></pre>\n", | |
| "<pre class='cython code score-2 '> __pyx_t_5 = <span class='pyx_c_api'>__Pyx_PyObject_to_MemoryviewSlice_ds_double</span>(__pyx_v_inarr);\n", | |
| " if (unlikely(!__pyx_t_5.memview)) __PYX_ERR(0, 6, __pyx_L1_error)\n", | |
| " __pyx_v_inarr_memview = __pyx_t_5;\n", | |
| " __pyx_t_5.memview = NULL;\n", | |
| " __pyx_t_5.data = NULL;\n", | |
| "</pre><pre class=\"cython line score-0\"> <span class=\"\">07</span>: </pre>\n", | |
| "<pre class=\"cython line score-28\" onclick='toggleDiv(this)'>+<span class=\"\">08</span>: <span class=\"n\">c_fast_addmul</span><span class=\"p\">(</span><span class=\"n\">inarr_memview</span><span class=\"p\">,</span> <span class=\"n\">outarr</span><span class=\"p\">,</span> <span class=\"nb\">len</span><span class=\"p\">(</span><span class=\"n\">inarr</span><span class=\"p\">),</span> <span class=\"n\">toadd</span><span class=\"p\">,</span> <span class=\"n\">tomul</span><span class=\"p\">)</span></pre>\n", | |
| "<pre class='cython code score-28 '> __pyx_t_5 = <span class='pyx_c_api'>__Pyx_PyObject_to_MemoryviewSlice_ds_double</span>(__pyx_v_outarr);\n", | |
| " if (unlikely(!__pyx_t_5.memview)) __PYX_ERR(0, 8, __pyx_L1_error)\n", | |
| " __pyx_t_6 = <span class='py_c_api'>PyObject_Length</span>(__pyx_v_inarr); if (unlikely(__pyx_t_6 == -1)) __PYX_ERR(0, 8, __pyx_L1_error)\n", | |
| " __pyx_t_7 = __pyx_<span class='py_c_api'>PyFloat_AsDouble</span>(__pyx_v_toadd); if (unlikely((__pyx_t_7 == (double)-1) && <span class='py_c_api'>PyErr_Occurred</span>())) __PYX_ERR(0, 8, __pyx_L1_error)\n", | |
| " __pyx_t_8 = __pyx_<span class='py_c_api'>PyFloat_AsDouble</span>(__pyx_v_tomul); if (unlikely((__pyx_t_8 == (double)-1) && <span class='py_c_api'>PyErr_Occurred</span>())) __PYX_ERR(0, 8, __pyx_L1_error)\n", | |
| " __pyx_t_1 = __pyx_f_46_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1_c_fast_addmul(__pyx_v_inarr_memview, __pyx_t_5, __pyx_t_6, __pyx_t_7, __pyx_t_8); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 8, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " __PYX_XDEC_MEMVIEW(&__pyx_t_5, 1);\n", | |
| " __pyx_t_5.memview = NULL;\n", | |
| " __pyx_t_5.data = NULL;\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_1); __pyx_t_1 = 0;\n", | |
| "</pre><pre class=\"cython line score-0\"> <span class=\"\">09</span>: </pre>\n", | |
| "<pre class=\"cython line score-2\" onclick='toggleDiv(this)'>+<span class=\"\">10</span>: <span class=\"k\">return</span> <span class=\"n\">outarr</span></pre>\n", | |
| "<pre class='cython code score-2 '> <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_r);\n", | |
| " <span class='pyx_macro_api'>__Pyx_INCREF</span>(__pyx_v_outarr);\n", | |
| " __pyx_r = __pyx_v_outarr;\n", | |
| " goto __pyx_L0;\n", | |
| "</pre><pre class=\"cython line score-0\"> <span class=\"\">11</span>: </pre>\n", | |
| "<pre class=\"cython line score-3\" onclick='toggleDiv(this)'>+<span class=\"\">12</span>: <span class=\"k\">cdef</span> <span class=\"nf\">c_fast_addmul</span><span class=\"p\">(</span><span class=\"n\">double</span><span class=\"p\">[:]</span> <span class=\"n\">inarr</span><span class=\"p\">,</span> <span class=\"n\">double</span><span class=\"p\">[:]</span> <span class=\"n\">outarr</span><span class=\"p\">,</span> <span class=\"nb\">int</span> <span class=\"n\">arrlen</span><span class=\"p\">,</span> <span class=\"n\">double</span> <span class=\"n\">toadd</span><span class=\"p\">,</span> <span class=\"n\">double</span> <span class=\"n\">tomul</span><span class=\"p\">):</span></pre>\n", | |
| "<pre class='cython code score-3 '>static PyObject *__pyx_f_46_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1_c_fast_addmul(__Pyx_memviewslice __pyx_v_inarr, __Pyx_memviewslice __pyx_v_outarr, int __pyx_v_arrlen, double __pyx_v_toadd, double __pyx_v_tomul) {\n", | |
| " int __pyx_v_i;\n", | |
| " PyObject *__pyx_r = NULL;\n", | |
| " <span class='refnanny'>__Pyx_RefNannyDeclarations</span>\n", | |
| " <span class='refnanny'>__Pyx_RefNannySetupContext</span>(\"c_fast_addmul\", 0);\n", | |
| "/* … */\n", | |
| " /* function exit code */\n", | |
| " __pyx_r = Py_None; <span class='pyx_macro_api'>__Pyx_INCREF</span>(Py_None);\n", | |
| " goto __pyx_L0;\n", | |
| " __pyx_L1_error:;\n", | |
| " <span class='pyx_c_api'>__Pyx_AddTraceback</span>(\"_cython_magic_d6ed5f1fc7de3b51bb80d535f0b73df1.c_fast_addmul\", __pyx_clineno, __pyx_lineno, __pyx_filename);\n", | |
| " __pyx_r = 0;\n", | |
| " __pyx_L0:;\n", | |
| " <span class='refnanny'>__Pyx_XGIVEREF</span>(__pyx_r);\n", | |
| " <span class='refnanny'>__Pyx_RefNannyFinishContext</span>();\n", | |
| " return __pyx_r;\n", | |
| "}\n", | |
| "</pre><pre class=\"cython line score-0\"> <span class=\"\">13</span>: <span class=\"k\">cdef</span> <span class=\"kt\">int</span> <span class=\"nf\">i</span></pre>\n", | |
| "<pre class=\"cython line score-0\" onclick='toggleDiv(this)'>+<span class=\"\">14</span>: <span class=\"k\">for</span> <span class=\"n\">i</span> <span class=\"ow\">in</span> <span class=\"nb\">range</span><span class=\"p\">(</span><span class=\"n\">arrlen</span><span class=\"p\">):</span></pre>\n", | |
| "<pre class='cython code score-0 '> __pyx_t_1 = __pyx_v_arrlen;\n", | |
| " for (__pyx_t_2 = 0; __pyx_t_2 < __pyx_t_1; __pyx_t_2+=1) {\n", | |
| " __pyx_v_i = __pyx_t_2;\n", | |
| "</pre><pre class=\"cython line score-4\" onclick='toggleDiv(this)'>+<span class=\"\">15</span>: <span class=\"n\">outarr</span><span class=\"p\">[</span><span class=\"n\">i</span><span class=\"p\">]</span> <span class=\"o\">=</span> <span class=\"n\">inarr</span><span class=\"p\">[</span><span class=\"n\">i</span><span class=\"p\">]</span><span class=\"o\">*</span><span class=\"n\">tomul</span> <span class=\"o\">+</span> <span class=\"n\">toadd</span></pre>\n", | |
| "<pre class='cython code score-4 '> __pyx_t_3 = __pyx_v_i;\n", | |
| " __pyx_t_4 = -1;\n", | |
| " if (__pyx_t_3 < 0) {\n", | |
| " __pyx_t_3 += __pyx_v_inarr.shape[0];\n", | |
| " if (unlikely(__pyx_t_3 < 0)) __pyx_t_4 = 0;\n", | |
| " } else if (unlikely(__pyx_t_3 >= __pyx_v_inarr.shape[0])) __pyx_t_4 = 0;\n", | |
| " if (unlikely(__pyx_t_4 != -1)) {\n", | |
| " <span class='pyx_c_api'>__Pyx_RaiseBufferIndexError</span>(__pyx_t_4);\n", | |
| " __PYX_ERR(0, 15, __pyx_L1_error)\n", | |
| " }\n", | |
| " __pyx_t_5 = __pyx_v_i;\n", | |
| " __pyx_t_4 = -1;\n", | |
| " if (__pyx_t_5 < 0) {\n", | |
| " __pyx_t_5 += __pyx_v_outarr.shape[0];\n", | |
| " if (unlikely(__pyx_t_5 < 0)) __pyx_t_4 = 0;\n", | |
| " } else if (unlikely(__pyx_t_5 >= __pyx_v_outarr.shape[0])) __pyx_t_4 = 0;\n", | |
| " if (unlikely(__pyx_t_4 != -1)) {\n", | |
| " <span class='pyx_c_api'>__Pyx_RaiseBufferIndexError</span>(__pyx_t_4);\n", | |
| " __PYX_ERR(0, 15, __pyx_L1_error)\n", | |
| " }\n", | |
| " *((double *) ( /* dim=0 */ (__pyx_v_outarr.data + __pyx_t_5 * __pyx_v_outarr.strides[0]) )) = (((*((double *) ( /* dim=0 */ (__pyx_v_inarr.data + __pyx_t_3 * __pyx_v_inarr.strides[0]) ))) * __pyx_v_tomul) + __pyx_v_toadd);\n", | |
| " }\n", | |
| "</pre></div></body></html>" | |
| ], | |
| "text/plain": [ | |
| "<IPython.core.display.HTML object>" | |
| ] | |
| }, | |
| "execution_count": 10, | |
| "metadata": {}, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "%%cython -a\n", | |
| "\n", | |
| "import numpy as np\n", | |
| "\n", | |
| "def fast_addmul(inarr, tomul, toadd):\n", | |
| " outarr = np.empty_like(inarr)\n", | |
| " cdef double[:] inarr_memview = inarr\n", | |
| " \n", | |
| " c_fast_addmul(inarr_memview, outarr, len(inarr), toadd, tomul)\n", | |
| " \n", | |
| " return outarr\n", | |
| "\n", | |
| "cdef c_fast_addmul(double[:] inarr, double[:] outarr, int arrlen, double toadd, double tomul):\n", | |
| " cdef int i\n", | |
| " for i in range(arrlen):\n", | |
| " outarr[i] = inarr[i]*tomul + toadd" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 11, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "data": { | |
| "text/html": [ | |
| "<!DOCTYPE html>\n", | |
| "<!-- Generated by Cython 0.25.2 -->\n", | |
| "<html>\n", | |
| "<head>\n", | |
| " <meta http-equiv=\"Content-Type\" content=\"text/html; charset=utf-8\" />\n", | |
| " <title>Cython: _cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5.pyx</title>\n", | |
| " <style type=\"text/css\">\n", | |
| " \n", | |
| "body.cython { font-family: courier; font-size: 12; }\n", | |
| "\n", | |
| ".cython.tag { }\n", | |
| ".cython.line { margin: 0em }\n", | |
| ".cython.code { font-size: 9; color: #444444; display: none; margin: 0px 0px 0px 8px; border-left: 8px none; }\n", | |
| "\n", | |
| ".cython.line .run { background-color: #B0FFB0; }\n", | |
| ".cython.line .mis { background-color: #FFB0B0; }\n", | |
| ".cython.code.run { border-left: 8px solid #B0FFB0; }\n", | |
| ".cython.code.mis { border-left: 8px solid #FFB0B0; }\n", | |
| "\n", | |
| ".cython.code .py_c_api { color: red; }\n", | |
| ".cython.code .py_macro_api { color: #FF7000; }\n", | |
| ".cython.code .pyx_c_api { color: #FF3000; }\n", | |
| ".cython.code .pyx_macro_api { color: #FF7000; }\n", | |
| ".cython.code .refnanny { color: #FFA000; }\n", | |
| ".cython.code .trace { color: #FFA000; }\n", | |
| ".cython.code .error_goto { color: #FFA000; }\n", | |
| "\n", | |
| ".cython.code .coerce { color: #008000; border: 1px dotted #008000 }\n", | |
| ".cython.code .py_attr { color: #FF0000; font-weight: bold; }\n", | |
| ".cython.code .c_attr { color: #0000FF; }\n", | |
| ".cython.code .py_call { color: #FF0000; font-weight: bold; }\n", | |
| ".cython.code .c_call { color: #0000FF; }\n", | |
| "\n", | |
| ".cython.score-0 {background-color: #FFFFff;}\n", | |
| ".cython.score-1 {background-color: #FFFFe7;}\n", | |
| ".cython.score-2 {background-color: #FFFFd4;}\n", | |
| ".cython.score-3 {background-color: #FFFFc4;}\n", | |
| ".cython.score-4 {background-color: #FFFFb6;}\n", | |
| ".cython.score-5 {background-color: #FFFFaa;}\n", | |
| ".cython.score-6 {background-color: #FFFF9f;}\n", | |
| ".cython.score-7 {background-color: #FFFF96;}\n", | |
| ".cython.score-8 {background-color: #FFFF8d;}\n", | |
| ".cython.score-9 {background-color: #FFFF86;}\n", | |
| ".cython.score-10 {background-color: #FFFF7f;}\n", | |
| ".cython.score-11 {background-color: #FFFF79;}\n", | |
| ".cython.score-12 {background-color: #FFFF73;}\n", | |
| ".cython.score-13 {background-color: #FFFF6e;}\n", | |
| ".cython.score-14 {background-color: #FFFF6a;}\n", | |
| ".cython.score-15 {background-color: #FFFF66;}\n", | |
| ".cython.score-16 {background-color: #FFFF62;}\n", | |
| ".cython.score-17 {background-color: #FFFF5e;}\n", | |
| ".cython.score-18 {background-color: #FFFF5b;}\n", | |
| ".cython.score-19 {background-color: #FFFF57;}\n", | |
| ".cython.score-20 {background-color: #FFFF55;}\n", | |
| ".cython.score-21 {background-color: #FFFF52;}\n", | |
| ".cython.score-22 {background-color: #FFFF4f;}\n", | |
| ".cython.score-23 {background-color: #FFFF4d;}\n", | |
| ".cython.score-24 {background-color: #FFFF4b;}\n", | |
| ".cython.score-25 {background-color: #FFFF48;}\n", | |
| ".cython.score-26 {background-color: #FFFF46;}\n", | |
| ".cython.score-27 {background-color: #FFFF44;}\n", | |
| ".cython.score-28 {background-color: #FFFF43;}\n", | |
| ".cython.score-29 {background-color: #FFFF41;}\n", | |
| ".cython.score-30 {background-color: #FFFF3f;}\n", | |
| ".cython.score-31 {background-color: #FFFF3e;}\n", | |
| ".cython.score-32 {background-color: #FFFF3c;}\n", | |
| ".cython.score-33 {background-color: #FFFF3b;}\n", | |
| ".cython.score-34 {background-color: #FFFF39;}\n", | |
| ".cython.score-35 {background-color: #FFFF38;}\n", | |
| ".cython.score-36 {background-color: #FFFF37;}\n", | |
| ".cython.score-37 {background-color: #FFFF36;}\n", | |
| ".cython.score-38 {background-color: #FFFF35;}\n", | |
| ".cython.score-39 {background-color: #FFFF34;}\n", | |
| ".cython.score-40 {background-color: #FFFF33;}\n", | |
| ".cython.score-41 {background-color: #FFFF32;}\n", | |
| ".cython.score-42 {background-color: #FFFF31;}\n", | |
| ".cython.score-43 {background-color: #FFFF30;}\n", | |
| ".cython.score-44 {background-color: #FFFF2f;}\n", | |
| ".cython.score-45 {background-color: #FFFF2e;}\n", | |
| ".cython.score-46 {background-color: #FFFF2d;}\n", | |
| ".cython.score-47 {background-color: #FFFF2c;}\n", | |
| ".cython.score-48 {background-color: #FFFF2b;}\n", | |
| ".cython.score-49 {background-color: #FFFF2b;}\n", | |
| ".cython.score-50 {background-color: #FFFF2a;}\n", | |
| ".cython.score-51 {background-color: #FFFF29;}\n", | |
| ".cython.score-52 {background-color: #FFFF29;}\n", | |
| ".cython.score-53 {background-color: #FFFF28;}\n", | |
| ".cython.score-54 {background-color: #FFFF27;}\n", | |
| ".cython.score-55 {background-color: #FFFF27;}\n", | |
| ".cython.score-56 {background-color: #FFFF26;}\n", | |
| ".cython.score-57 {background-color: #FFFF26;}\n", | |
| ".cython.score-58 {background-color: #FFFF25;}\n", | |
| ".cython.score-59 {background-color: #FFFF24;}\n", | |
| ".cython.score-60 {background-color: #FFFF24;}\n", | |
| ".cython.score-61 {background-color: #FFFF23;}\n", | |
| ".cython.score-62 {background-color: #FFFF23;}\n", | |
| ".cython.score-63 {background-color: #FFFF22;}\n", | |
| ".cython.score-64 {background-color: #FFFF22;}\n", | |
| ".cython.score-65 {background-color: #FFFF22;}\n", | |
| ".cython.score-66 {background-color: #FFFF21;}\n", | |
| ".cython.score-67 {background-color: #FFFF21;}\n", | |
| ".cython.score-68 {background-color: #FFFF20;}\n", | |
| ".cython.score-69 {background-color: #FFFF20;}\n", | |
| ".cython.score-70 {background-color: #FFFF1f;}\n", | |
| ".cython.score-71 {background-color: #FFFF1f;}\n", | |
| ".cython.score-72 {background-color: #FFFF1f;}\n", | |
| ".cython.score-73 {background-color: #FFFF1e;}\n", | |
| ".cython.score-74 {background-color: #FFFF1e;}\n", | |
| ".cython.score-75 {background-color: #FFFF1e;}\n", | |
| ".cython.score-76 {background-color: #FFFF1d;}\n", | |
| ".cython.score-77 {background-color: #FFFF1d;}\n", | |
| ".cython.score-78 {background-color: #FFFF1c;}\n", | |
| ".cython.score-79 {background-color: #FFFF1c;}\n", | |
| ".cython.score-80 {background-color: #FFFF1c;}\n", | |
| ".cython.score-81 {background-color: #FFFF1c;}\n", | |
| ".cython.score-82 {background-color: #FFFF1b;}\n", | |
| ".cython.score-83 {background-color: #FFFF1b;}\n", | |
| ".cython.score-84 {background-color: #FFFF1b;}\n", | |
| ".cython.score-85 {background-color: #FFFF1a;}\n", | |
| ".cython.score-86 {background-color: #FFFF1a;}\n", | |
| ".cython.score-87 {background-color: #FFFF1a;}\n", | |
| ".cython.score-88 {background-color: #FFFF1a;}\n", | |
| ".cython.score-89 {background-color: #FFFF19;}\n", | |
| ".cython.score-90 {background-color: #FFFF19;}\n", | |
| ".cython.score-91 {background-color: #FFFF19;}\n", | |
| ".cython.score-92 {background-color: #FFFF19;}\n", | |
| ".cython.score-93 {background-color: #FFFF18;}\n", | |
| ".cython.score-94 {background-color: #FFFF18;}\n", | |
| ".cython.score-95 {background-color: #FFFF18;}\n", | |
| ".cython.score-96 {background-color: #FFFF18;}\n", | |
| ".cython.score-97 {background-color: #FFFF17;}\n", | |
| ".cython.score-98 {background-color: #FFFF17;}\n", | |
| ".cython.score-99 {background-color: #FFFF17;}\n", | |
| ".cython.score-100 {background-color: #FFFF17;}\n", | |
| ".cython.score-101 {background-color: #FFFF16;}\n", | |
| ".cython.score-102 {background-color: #FFFF16;}\n", | |
| ".cython.score-103 {background-color: #FFFF16;}\n", | |
| ".cython.score-104 {background-color: #FFFF16;}\n", | |
| ".cython.score-105 {background-color: #FFFF16;}\n", | |
| ".cython.score-106 {background-color: #FFFF15;}\n", | |
| ".cython.score-107 {background-color: #FFFF15;}\n", | |
| ".cython.score-108 {background-color: #FFFF15;}\n", | |
| ".cython.score-109 {background-color: #FFFF15;}\n", | |
| ".cython.score-110 {background-color: #FFFF15;}\n", | |
| ".cython.score-111 {background-color: #FFFF15;}\n", | |
| ".cython.score-112 {background-color: #FFFF14;}\n", | |
| ".cython.score-113 {background-color: #FFFF14;}\n", | |
| ".cython.score-114 {background-color: #FFFF14;}\n", | |
| ".cython.score-115 {background-color: #FFFF14;}\n", | |
| ".cython.score-116 {background-color: #FFFF14;}\n", | |
| ".cython.score-117 {background-color: #FFFF14;}\n", | |
| ".cython.score-118 {background-color: #FFFF13;}\n", | |
| ".cython.score-119 {background-color: #FFFF13;}\n", | |
| ".cython.score-120 {background-color: #FFFF13;}\n", | |
| ".cython.score-121 {background-color: #FFFF13;}\n", | |
| ".cython.score-122 {background-color: #FFFF13;}\n", | |
| ".cython.score-123 {background-color: #FFFF13;}\n", | |
| ".cython.score-124 {background-color: #FFFF13;}\n", | |
| ".cython.score-125 {background-color: #FFFF12;}\n", | |
| ".cython.score-126 {background-color: #FFFF12;}\n", | |
| ".cython.score-127 {background-color: #FFFF12;}\n", | |
| ".cython.score-128 {background-color: #FFFF12;}\n", | |
| ".cython.score-129 {background-color: #FFFF12;}\n", | |
| ".cython.score-130 {background-color: #FFFF12;}\n", | |
| ".cython.score-131 {background-color: #FFFF12;}\n", | |
| ".cython.score-132 {background-color: #FFFF11;}\n", | |
| ".cython.score-133 {background-color: #FFFF11;}\n", | |
| ".cython.score-134 {background-color: #FFFF11;}\n", | |
| ".cython.score-135 {background-color: #FFFF11;}\n", | |
| ".cython.score-136 {background-color: #FFFF11;}\n", | |
| ".cython.score-137 {background-color: #FFFF11;}\n", | |
| ".cython.score-138 {background-color: #FFFF11;}\n", | |
| ".cython.score-139 {background-color: #FFFF11;}\n", | |
| ".cython.score-140 {background-color: #FFFF11;}\n", | |
| ".cython.score-141 {background-color: #FFFF10;}\n", | |
| ".cython.score-142 {background-color: #FFFF10;}\n", | |
| ".cython.score-143 {background-color: #FFFF10;}\n", | |
| ".cython.score-144 {background-color: #FFFF10;}\n", | |
| ".cython.score-145 {background-color: #FFFF10;}\n", | |
| ".cython.score-146 {background-color: #FFFF10;}\n", | |
| ".cython.score-147 {background-color: #FFFF10;}\n", | |
| ".cython.score-148 {background-color: #FFFF10;}\n", | |
| ".cython.score-149 {background-color: #FFFF10;}\n", | |
| ".cython.score-150 {background-color: #FFFF0f;}\n", | |
| ".cython.score-151 {background-color: #FFFF0f;}\n", | |
| ".cython.score-152 {background-color: #FFFF0f;}\n", | |
| ".cython.score-153 {background-color: #FFFF0f;}\n", | |
| ".cython.score-154 {background-color: #FFFF0f;}\n", | |
| ".cython.score-155 {background-color: #FFFF0f;}\n", | |
| ".cython.score-156 {background-color: #FFFF0f;}\n", | |
| ".cython.score-157 {background-color: #FFFF0f;}\n", | |
| ".cython.score-158 {background-color: #FFFF0f;}\n", | |
| ".cython.score-159 {background-color: #FFFF0f;}\n", | |
| ".cython.score-160 {background-color: #FFFF0f;}\n", | |
| ".cython.score-161 {background-color: #FFFF0e;}\n", | |
| ".cython.score-162 {background-color: #FFFF0e;}\n", | |
| ".cython.score-163 {background-color: #FFFF0e;}\n", | |
| ".cython.score-164 {background-color: #FFFF0e;}\n", | |
| ".cython.score-165 {background-color: #FFFF0e;}\n", | |
| ".cython.score-166 {background-color: #FFFF0e;}\n", | |
| ".cython.score-167 {background-color: #FFFF0e;}\n", | |
| ".cython.score-168 {background-color: #FFFF0e;}\n", | |
| ".cython.score-169 {background-color: #FFFF0e;}\n", | |
| ".cython.score-170 {background-color: #FFFF0e;}\n", | |
| ".cython.score-171 {background-color: #FFFF0e;}\n", | |
| ".cython.score-172 {background-color: #FFFF0e;}\n", | |
| ".cython.score-173 {background-color: #FFFF0d;}\n", | |
| ".cython.score-174 {background-color: #FFFF0d;}\n", | |
| ".cython.score-175 {background-color: #FFFF0d;}\n", | |
| ".cython.score-176 {background-color: #FFFF0d;}\n", | |
| ".cython.score-177 {background-color: #FFFF0d;}\n", | |
| ".cython.score-178 {background-color: #FFFF0d;}\n", | |
| ".cython.score-179 {background-color: #FFFF0d;}\n", | |
| ".cython.score-180 {background-color: #FFFF0d;}\n", | |
| ".cython.score-181 {background-color: #FFFF0d;}\n", | |
| ".cython.score-182 {background-color: #FFFF0d;}\n", | |
| ".cython.score-183 {background-color: #FFFF0d;}\n", | |
| ".cython.score-184 {background-color: #FFFF0d;}\n", | |
| ".cython.score-185 {background-color: #FFFF0d;}\n", | |
| ".cython.score-186 {background-color: #FFFF0d;}\n", | |
| ".cython.score-187 {background-color: #FFFF0c;}\n", | |
| ".cython.score-188 {background-color: #FFFF0c;}\n", | |
| ".cython.score-189 {background-color: #FFFF0c;}\n", | |
| ".cython.score-190 {background-color: #FFFF0c;}\n", | |
| ".cython.score-191 {background-color: #FFFF0c;}\n", | |
| ".cython.score-192 {background-color: #FFFF0c;}\n", | |
| ".cython.score-193 {background-color: #FFFF0c;}\n", | |
| ".cython.score-194 {background-color: #FFFF0c;}\n", | |
| ".cython.score-195 {background-color: #FFFF0c;}\n", | |
| ".cython.score-196 {background-color: #FFFF0c;}\n", | |
| ".cython.score-197 {background-color: #FFFF0c;}\n", | |
| ".cython.score-198 {background-color: #FFFF0c;}\n", | |
| ".cython.score-199 {background-color: #FFFF0c;}\n", | |
| ".cython.score-200 {background-color: #FFFF0c;}\n", | |
| ".cython.score-201 {background-color: #FFFF0c;}\n", | |
| ".cython.score-202 {background-color: #FFFF0c;}\n", | |
| ".cython.score-203 {background-color: #FFFF0b;}\n", | |
| ".cython.score-204 {background-color: #FFFF0b;}\n", | |
| ".cython.score-205 {background-color: #FFFF0b;}\n", | |
| ".cython.score-206 {background-color: #FFFF0b;}\n", | |
| ".cython.score-207 {background-color: #FFFF0b;}\n", | |
| ".cython.score-208 {background-color: #FFFF0b;}\n", | |
| ".cython.score-209 {background-color: #FFFF0b;}\n", | |
| ".cython.score-210 {background-color: #FFFF0b;}\n", | |
| ".cython.score-211 {background-color: #FFFF0b;}\n", | |
| ".cython.score-212 {background-color: #FFFF0b;}\n", | |
| ".cython.score-213 {background-color: #FFFF0b;}\n", | |
| ".cython.score-214 {background-color: #FFFF0b;}\n", | |
| ".cython.score-215 {background-color: #FFFF0b;}\n", | |
| ".cython.score-216 {background-color: #FFFF0b;}\n", | |
| ".cython.score-217 {background-color: #FFFF0b;}\n", | |
| ".cython.score-218 {background-color: #FFFF0b;}\n", | |
| ".cython.score-219 {background-color: #FFFF0b;}\n", | |
| ".cython.score-220 {background-color: #FFFF0b;}\n", | |
| ".cython.score-221 {background-color: #FFFF0b;}\n", | |
| ".cython.score-222 {background-color: #FFFF0a;}\n", | |
| ".cython.score-223 {background-color: #FFFF0a;}\n", | |
| ".cython.score-224 {background-color: #FFFF0a;}\n", | |
| ".cython.score-225 {background-color: #FFFF0a;}\n", | |
| ".cython.score-226 {background-color: #FFFF0a;}\n", | |
| ".cython.score-227 {background-color: #FFFF0a;}\n", | |
| ".cython.score-228 {background-color: #FFFF0a;}\n", | |
| ".cython.score-229 {background-color: #FFFF0a;}\n", | |
| ".cython.score-230 {background-color: #FFFF0a;}\n", | |
| ".cython.score-231 {background-color: #FFFF0a;}\n", | |
| ".cython.score-232 {background-color: #FFFF0a;}\n", | |
| ".cython.score-233 {background-color: #FFFF0a;}\n", | |
| ".cython.score-234 {background-color: #FFFF0a;}\n", | |
| ".cython.score-235 {background-color: #FFFF0a;}\n", | |
| ".cython.score-236 {background-color: #FFFF0a;}\n", | |
| ".cython.score-237 {background-color: #FFFF0a;}\n", | |
| ".cython.score-238 {background-color: #FFFF0a;}\n", | |
| ".cython.score-239 {background-color: #FFFF0a;}\n", | |
| ".cython.score-240 {background-color: #FFFF0a;}\n", | |
| ".cython.score-241 {background-color: #FFFF0a;}\n", | |
| ".cython.score-242 {background-color: #FFFF0a;}\n", | |
| ".cython.score-243 {background-color: #FFFF0a;}\n", | |
| ".cython.score-244 {background-color: #FFFF0a;}\n", | |
| ".cython.score-245 {background-color: #FFFF0a;}\n", | |
| ".cython.score-246 {background-color: #FFFF09;}\n", | |
| ".cython.score-247 {background-color: #FFFF09;}\n", | |
| ".cython.score-248 {background-color: #FFFF09;}\n", | |
| ".cython.score-249 {background-color: #FFFF09;}\n", | |
| ".cython.score-250 {background-color: #FFFF09;}\n", | |
| ".cython.score-251 {background-color: #FFFF09;}\n", | |
| ".cython.score-252 {background-color: #FFFF09;}\n", | |
| ".cython.score-253 {background-color: #FFFF09;}\n", | |
| ".cython.score-254 {background-color: #FFFF09;}\n", | |
| ".cython .hll { background-color: #ffffcc }\n", | |
| ".cython { background: #f8f8f8; }\n", | |
| ".cython .c { color: #408080; font-style: italic } /* Comment */\n", | |
| ".cython .err { border: 1px solid #FF0000 } /* Error */\n", | |
| ".cython .k { color: #008000; font-weight: bold } /* Keyword */\n", | |
| ".cython .o { color: #666666 } /* Operator */\n", | |
| ".cython .ch { color: #408080; font-style: italic } /* Comment.Hashbang */\n", | |
| ".cython .cm { color: #408080; font-style: italic } /* Comment.Multiline */\n", | |
| ".cython .cp { color: #BC7A00 } /* Comment.Preproc */\n", | |
| ".cython .cpf { color: #408080; font-style: italic } /* Comment.PreprocFile */\n", | |
| ".cython .c1 { color: #408080; font-style: italic } /* Comment.Single */\n", | |
| ".cython .cs { color: #408080; font-style: italic } /* Comment.Special */\n", | |
| ".cython .gd { color: #A00000 } /* Generic.Deleted */\n", | |
| ".cython .ge { font-style: italic } /* Generic.Emph */\n", | |
| ".cython .gr { color: #FF0000 } /* Generic.Error */\n", | |
| ".cython .gh { color: #000080; font-weight: bold } /* Generic.Heading */\n", | |
| ".cython .gi { color: #00A000 } /* Generic.Inserted */\n", | |
| ".cython .go { color: #888888 } /* Generic.Output */\n", | |
| ".cython .gp { color: #000080; font-weight: bold } /* Generic.Prompt */\n", | |
| ".cython .gs { font-weight: bold } /* Generic.Strong */\n", | |
| ".cython .gu { color: #800080; font-weight: bold } /* Generic.Subheading */\n", | |
| ".cython .gt { color: #0044DD } /* Generic.Traceback */\n", | |
| ".cython .kc { color: #008000; font-weight: bold } /* Keyword.Constant */\n", | |
| ".cython .kd { color: #008000; font-weight: bold } /* Keyword.Declaration */\n", | |
| ".cython .kn { color: #008000; font-weight: bold } /* Keyword.Namespace */\n", | |
| ".cython .kp { color: #008000 } /* Keyword.Pseudo */\n", | |
| ".cython .kr { color: #008000; font-weight: bold } /* Keyword.Reserved */\n", | |
| ".cython .kt { color: #B00040 } /* Keyword.Type */\n", | |
| ".cython .m { color: #666666 } /* Literal.Number */\n", | |
| ".cython .s { color: #BA2121 } /* Literal.String */\n", | |
| ".cython .na { color: #7D9029 } /* Name.Attribute */\n", | |
| ".cython .nb { color: #008000 } /* Name.Builtin */\n", | |
| ".cython .nc { color: #0000FF; font-weight: bold } /* Name.Class */\n", | |
| ".cython .no { color: #880000 } /* Name.Constant */\n", | |
| ".cython .nd { color: #AA22FF } /* Name.Decorator */\n", | |
| ".cython .ni { color: #999999; font-weight: bold } /* Name.Entity */\n", | |
| ".cython .ne { color: #D2413A; font-weight: bold } /* Name.Exception */\n", | |
| ".cython .nf { color: #0000FF } /* Name.Function */\n", | |
| ".cython .nl { color: #A0A000 } /* Name.Label */\n", | |
| ".cython .nn { color: #0000FF; font-weight: bold } /* Name.Namespace */\n", | |
| ".cython .nt { color: #008000; font-weight: bold } /* Name.Tag */\n", | |
| ".cython .nv { color: #19177C } /* Name.Variable */\n", | |
| ".cython .ow { color: #AA22FF; font-weight: bold } /* Operator.Word */\n", | |
| ".cython .w { color: #bbbbbb } /* Text.Whitespace */\n", | |
| ".cython .mb { color: #666666 } /* Literal.Number.Bin */\n", | |
| ".cython .mf { color: #666666 } /* Literal.Number.Float */\n", | |
| ".cython .mh { color: #666666 } /* Literal.Number.Hex */\n", | |
| ".cython .mi { color: #666666 } /* Literal.Number.Integer */\n", | |
| ".cython .mo { color: #666666 } /* Literal.Number.Oct */\n", | |
| ".cython .sb { color: #BA2121 } /* Literal.String.Backtick */\n", | |
| ".cython .sc { color: #BA2121 } /* Literal.String.Char */\n", | |
| ".cython .sd { color: #BA2121; font-style: italic } /* Literal.String.Doc */\n", | |
| ".cython .s2 { color: #BA2121 } /* Literal.String.Double */\n", | |
| ".cython .se { color: #BB6622; font-weight: bold } /* Literal.String.Escape */\n", | |
| ".cython .sh { color: #BA2121 } /* Literal.String.Heredoc */\n", | |
| ".cython .si { color: #BB6688; font-weight: bold } /* Literal.String.Interpol */\n", | |
| ".cython .sx { color: #008000 } /* Literal.String.Other */\n", | |
| ".cython .sr { color: #BB6688 } /* Literal.String.Regex */\n", | |
| ".cython .s1 { color: #BA2121 } /* Literal.String.Single */\n", | |
| ".cython .ss { color: #19177C } /* Literal.String.Symbol */\n", | |
| ".cython .bp { color: #008000 } /* Name.Builtin.Pseudo */\n", | |
| ".cython .vc { color: #19177C } /* Name.Variable.Class */\n", | |
| ".cython .vg { color: #19177C } /* Name.Variable.Global */\n", | |
| ".cython .vi { color: #19177C } /* Name.Variable.Instance */\n", | |
| ".cython .il { color: #666666 } /* Literal.Number.Integer.Long */\n", | |
| " </style>\n", | |
| " <script>\n", | |
| " function toggleDiv(id) {\n", | |
| " theDiv = id.nextElementSibling\n", | |
| " if (theDiv.style.display != 'block') theDiv.style.display = 'block';\n", | |
| " else theDiv.style.display = 'none';\n", | |
| " }\n", | |
| " </script>\n", | |
| "</head>\n", | |
| "<body class=\"cython\">\n", | |
| "<p><span style=\"border-bottom: solid 1px grey;\">Generated by Cython 0.25.2</span></p>\n", | |
| "<p>\n", | |
| " <span style=\"background-color: #FFFF00\">Yellow lines</span> hint at Python interaction.<br />\n", | |
| " Click on a line that starts with a \"<code>+</code>\" to see the C code that Cython generated for it.\n", | |
| "</p>\n", | |
| "<div class=\"cython\"><pre class=\"cython line score-0\"> <span class=\"\">01</span>: </pre>\n", | |
| "<pre class=\"cython line score-11\" onclick='toggleDiv(this)'>+<span class=\"\">02</span>: <span class=\"k\">import</span> <span class=\"nn\">cython</span></pre>\n", | |
| "<pre class='cython code score-11 '> __pyx_t_1 = <span class='py_c_api'>PyDict_New</span>(); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 2, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " if (<span class='py_c_api'>PyDict_SetItem</span>(__pyx_d, __pyx_n_s_test, __pyx_t_1) < 0) __PYX_ERR(0, 2, __pyx_L1_error)\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_1); __pyx_t_1 = 0;\n", | |
| "</pre><pre class=\"cython line score-8\" onclick='toggleDiv(this)'>+<span class=\"\">03</span>: <span class=\"k\">import</span> <span class=\"nn\">numpy</span> <span class=\"k\">as</span> <span class=\"nn\">np</span></pre>\n", | |
| "<pre class='cython code score-8 '> __pyx_t_1 = <span class='pyx_c_api'>__Pyx_Import</span>(__pyx_n_s_numpy, 0, 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 3, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " if (<span class='py_c_api'>PyDict_SetItem</span>(__pyx_d, __pyx_n_s_np, __pyx_t_1) < 0) __PYX_ERR(0, 3, __pyx_L1_error)\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_1); __pyx_t_1 = 0;\n", | |
| "</pre><pre class=\"cython line score-0\"> <span class=\"\">04</span>: </pre>\n", | |
| "<pre class=\"cython line score-59\" onclick='toggleDiv(this)'>+<span class=\"\">05</span>: <span class=\"k\">def</span> <span class=\"nf\">fast_addmul_opt</span><span class=\"p\">(</span><span class=\"n\">inarr</span><span class=\"p\">,</span> <span class=\"n\">tomul</span><span class=\"p\">,</span> <span class=\"n\">toadd</span><span class=\"p\">):</span></pre>\n", | |
| "<pre class='cython code score-59 '>/* Python wrapper */\n", | |
| "static PyObject *__pyx_pw_46_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5_1fast_addmul_opt(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/\n", | |
| "static PyMethodDef __pyx_mdef_46_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5_1fast_addmul_opt = {\"fast_addmul_opt\", (PyCFunction)__pyx_pw_46_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5_1fast_addmul_opt, METH_VARARGS|METH_KEYWORDS, 0};\n", | |
| "static PyObject *__pyx_pw_46_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5_1fast_addmul_opt(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {\n", | |
| " PyObject *__pyx_v_inarr = 0;\n", | |
| " PyObject *__pyx_v_tomul = 0;\n", | |
| " PyObject *__pyx_v_toadd = 0;\n", | |
| " PyObject *__pyx_r = 0;\n", | |
| " <span class='refnanny'>__Pyx_RefNannyDeclarations</span>\n", | |
| " <span class='refnanny'>__Pyx_RefNannySetupContext</span>(\"fast_addmul_opt (wrapper)\", 0);\n", | |
| " {\n", | |
| " static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_inarr,&__pyx_n_s_tomul,&__pyx_n_s_toadd,0};\n", | |
| " PyObject* values[3] = {0,0,0};\n", | |
| " if (unlikely(__pyx_kwds)) {\n", | |
| " Py_ssize_t kw_args;\n", | |
| " const Py_ssize_t pos_args = <span class='py_macro_api'>PyTuple_GET_SIZE</span>(__pyx_args);\n", | |
| " switch (pos_args) {\n", | |
| " case 3: values[2] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 2);\n", | |
| " case 2: values[1] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 1);\n", | |
| " case 1: values[0] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 0);\n", | |
| " case 0: break;\n", | |
| " default: goto __pyx_L5_argtuple_error;\n", | |
| " }\n", | |
| " kw_args = <span class='py_c_api'>PyDict_Size</span>(__pyx_kwds);\n", | |
| " switch (pos_args) {\n", | |
| " case 0:\n", | |
| " if (likely((values[0] = <span class='py_c_api'>PyDict_GetItem</span>(__pyx_kwds, __pyx_n_s_inarr)) != 0)) kw_args--;\n", | |
| " else goto __pyx_L5_argtuple_error;\n", | |
| " case 1:\n", | |
| " if (likely((values[1] = <span class='py_c_api'>PyDict_GetItem</span>(__pyx_kwds, __pyx_n_s_tomul)) != 0)) kw_args--;\n", | |
| " else {\n", | |
| " <span class='pyx_c_api'>__Pyx_RaiseArgtupleInvalid</span>(\"fast_addmul_opt\", 1, 3, 3, 1); __PYX_ERR(0, 5, __pyx_L3_error)\n", | |
| " }\n", | |
| " case 2:\n", | |
| " if (likely((values[2] = <span class='py_c_api'>PyDict_GetItem</span>(__pyx_kwds, __pyx_n_s_toadd)) != 0)) kw_args--;\n", | |
| " else {\n", | |
| " <span class='pyx_c_api'>__Pyx_RaiseArgtupleInvalid</span>(\"fast_addmul_opt\", 1, 3, 3, 2); __PYX_ERR(0, 5, __pyx_L3_error)\n", | |
| " }\n", | |
| " }\n", | |
| " if (unlikely(kw_args > 0)) {\n", | |
| " if (unlikely(<span class='pyx_c_api'>__Pyx_ParseOptionalKeywords</span>(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, \"fast_addmul_opt\") < 0)) __PYX_ERR(0, 5, __pyx_L3_error)\n", | |
| " }\n", | |
| " } else if (<span class='py_macro_api'>PyTuple_GET_SIZE</span>(__pyx_args) != 3) {\n", | |
| " goto __pyx_L5_argtuple_error;\n", | |
| " } else {\n", | |
| " values[0] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 0);\n", | |
| " values[1] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 1);\n", | |
| " values[2] = <span class='py_macro_api'>PyTuple_GET_ITEM</span>(__pyx_args, 2);\n", | |
| " }\n", | |
| " __pyx_v_inarr = values[0];\n", | |
| " __pyx_v_tomul = values[1];\n", | |
| " __pyx_v_toadd = values[2];\n", | |
| " }\n", | |
| " goto __pyx_L4_argument_unpacking_done;\n", | |
| " __pyx_L5_argtuple_error:;\n", | |
| " <span class='pyx_c_api'>__Pyx_RaiseArgtupleInvalid</span>(\"fast_addmul_opt\", 1, 3, 3, <span class='py_macro_api'>PyTuple_GET_SIZE</span>(__pyx_args)); __PYX_ERR(0, 5, __pyx_L3_error)\n", | |
| " __pyx_L3_error:;\n", | |
| " <span class='pyx_c_api'>__Pyx_AddTraceback</span>(\"_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5.fast_addmul_opt\", __pyx_clineno, __pyx_lineno, __pyx_filename);\n", | |
| " <span class='refnanny'>__Pyx_RefNannyFinishContext</span>();\n", | |
| " return NULL;\n", | |
| " __pyx_L4_argument_unpacking_done:;\n", | |
| " __pyx_r = __pyx_pf_46_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5_fast_addmul_opt(__pyx_self, __pyx_v_inarr, __pyx_v_tomul, __pyx_v_toadd);\n", | |
| "\n", | |
| " /* function exit code */\n", | |
| " <span class='refnanny'>__Pyx_RefNannyFinishContext</span>();\n", | |
| " return __pyx_r;\n", | |
| "}\n", | |
| "\n", | |
| "static PyObject *__pyx_pf_46_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5_fast_addmul_opt(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_inarr, PyObject *__pyx_v_tomul, PyObject *__pyx_v_toadd) {\n", | |
| " PyObject *__pyx_v_outarr = NULL;\n", | |
| " __Pyx_memviewslice __pyx_v_inarr_memview = { 0, 0, { 0 }, { 0 }, { 0 } };\n", | |
| " PyObject *__pyx_r = NULL;\n", | |
| " <span class='refnanny'>__Pyx_RefNannyDeclarations</span>\n", | |
| " <span class='refnanny'>__Pyx_RefNannySetupContext</span>(\"fast_addmul_opt\", 0);\n", | |
| "/* … */\n", | |
| " /* function exit code */\n", | |
| " __pyx_L1_error:;\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_1);\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_2);\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_3);\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_4);\n", | |
| " __PYX_XDEC_MEMVIEW(&__pyx_t_5, 1);\n", | |
| " <span class='pyx_c_api'>__Pyx_AddTraceback</span>(\"_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5.fast_addmul_opt\", __pyx_clineno, __pyx_lineno, __pyx_filename);\n", | |
| " __pyx_r = NULL;\n", | |
| " __pyx_L0:;\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_v_outarr);\n", | |
| " __PYX_XDEC_MEMVIEW(&__pyx_v_inarr_memview, 1);\n", | |
| " <span class='refnanny'>__Pyx_XGIVEREF</span>(__pyx_r);\n", | |
| " <span class='refnanny'>__Pyx_RefNannyFinishContext</span>();\n", | |
| " return __pyx_r;\n", | |
| "}\n", | |
| "/* … */\n", | |
| " __pyx_tuple__14 = <span class='py_c_api'>PyTuple_Pack</span>(5, __pyx_n_s_inarr, __pyx_n_s_tomul, __pyx_n_s_toadd, __pyx_n_s_outarr, __pyx_n_s_inarr_memview); if (unlikely(!__pyx_tuple__14)) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_tuple__14);\n", | |
| " <span class='refnanny'>__Pyx_GIVEREF</span>(__pyx_tuple__14);\n", | |
| "/* … */\n", | |
| " __pyx_t_1 = PyCFunction_NewEx(&__pyx_mdef_46_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5_1fast_addmul_opt, NULL, __pyx_n_s_cython_magic_eb169e70499a2d4a8f); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " if (<span class='py_c_api'>PyDict_SetItem</span>(__pyx_d, __pyx_n_s_fast_addmul_opt, __pyx_t_1) < 0) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_1); __pyx_t_1 = 0;\n", | |
| " __pyx_codeobj__15 = (PyObject*)<span class='pyx_c_api'>__Pyx_PyCode_New</span>(3, 0, 5, 0, 0, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__14, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_Users_erik_ipython_cython__cyth, __pyx_n_s_fast_addmul_opt, 5, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__15)) __PYX_ERR(0, 5, __pyx_L1_error)\n", | |
| "</pre><pre class=\"cython line score-42\" onclick='toggleDiv(this)'>+<span class=\"\">06</span>: <span class=\"n\">outarr</span> <span class=\"o\">=</span> <span class=\"n\">np</span><span class=\"o\">.</span><span class=\"n\">empty_like</span><span class=\"p\">(</span><span class=\"n\">inarr</span><span class=\"p\">)</span></pre>\n", | |
| "<pre class='cython code score-42 '> __pyx_t_2 = <span class='pyx_c_api'>__Pyx_GetModuleGlobalName</span>(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 6, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_2);\n", | |
| " __pyx_t_3 = <span class='pyx_c_api'>__Pyx_PyObject_GetAttrStr</span>(__pyx_t_2, __pyx_n_s_empty_like); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 6, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_3);\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_2); __pyx_t_2 = 0;\n", | |
| " __pyx_t_2 = NULL;\n", | |
| " if (CYTHON_UNPACK_METHODS && unlikely(<span class='py_c_api'>PyMethod_Check</span>(__pyx_t_3))) {\n", | |
| " __pyx_t_2 = <span class='py_macro_api'>PyMethod_GET_SELF</span>(__pyx_t_3);\n", | |
| " if (likely(__pyx_t_2)) {\n", | |
| " PyObject* function = <span class='py_macro_api'>PyMethod_GET_FUNCTION</span>(__pyx_t_3);\n", | |
| " <span class='pyx_macro_api'>__Pyx_INCREF</span>(__pyx_t_2);\n", | |
| " <span class='pyx_macro_api'>__Pyx_INCREF</span>(function);\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF_SET</span>(__pyx_t_3, function);\n", | |
| " }\n", | |
| " }\n", | |
| " if (!__pyx_t_2) {\n", | |
| " __pyx_t_1 = <span class='pyx_c_api'>__Pyx_PyObject_CallOneArg</span>(__pyx_t_3, __pyx_v_inarr); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 6, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " } else {\n", | |
| " #if CYTHON_FAST_PYCALL\n", | |
| " if (<span class='py_c_api'>PyFunction_Check</span>(__pyx_t_3)) {\n", | |
| " PyObject *__pyx_temp[2] = {__pyx_t_2, __pyx_v_inarr};\n", | |
| " __pyx_t_1 = <span class='pyx_c_api'>__Pyx_PyFunction_FastCall</span>(__pyx_t_3, __pyx_temp+1-1, 1+1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 6, __pyx_L1_error)\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_2); __pyx_t_2 = 0;\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " } else\n", | |
| " #endif\n", | |
| " #if CYTHON_FAST_PYCCALL\n", | |
| " if (<span class='pyx_c_api'>__Pyx_PyFastCFunction_Check</span>(__pyx_t_3)) {\n", | |
| " PyObject *__pyx_temp[2] = {__pyx_t_2, __pyx_v_inarr};\n", | |
| " __pyx_t_1 = <span class='pyx_c_api'>__Pyx_PyCFunction_FastCall</span>(__pyx_t_3, __pyx_temp+1-1, 1+1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 6, __pyx_L1_error)\n", | |
| " <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_t_2); __pyx_t_2 = 0;\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " } else\n", | |
| " #endif\n", | |
| " {\n", | |
| " __pyx_t_4 = <span class='py_c_api'>PyTuple_New</span>(1+1); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 6, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_4);\n", | |
| " <span class='refnanny'>__Pyx_GIVEREF</span>(__pyx_t_2); <span class='py_macro_api'>PyTuple_SET_ITEM</span>(__pyx_t_4, 0, __pyx_t_2); __pyx_t_2 = NULL;\n", | |
| " <span class='pyx_macro_api'>__Pyx_INCREF</span>(__pyx_v_inarr);\n", | |
| " <span class='refnanny'>__Pyx_GIVEREF</span>(__pyx_v_inarr);\n", | |
| " <span class='py_macro_api'>PyTuple_SET_ITEM</span>(__pyx_t_4, 0+1, __pyx_v_inarr);\n", | |
| " __pyx_t_1 = <span class='pyx_c_api'>__Pyx_PyObject_Call</span>(__pyx_t_3, __pyx_t_4, NULL); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 6, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_4); __pyx_t_4 = 0;\n", | |
| " }\n", | |
| " }\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_3); __pyx_t_3 = 0;\n", | |
| " __pyx_v_outarr = __pyx_t_1;\n", | |
| " __pyx_t_1 = 0;\n", | |
| "</pre><pre class=\"cython line score-2\" onclick='toggleDiv(this)'>+<span class=\"\">07</span>: <span class=\"k\">cdef</span> <span class=\"kt\">double</span>[<span class=\"p\">:]</span> <span class=\"n\">inarr_memview</span> <span class=\"o\">=</span> <span class=\"n\">inarr</span></pre>\n", | |
| "<pre class='cython code score-2 '> __pyx_t_5 = <span class='pyx_c_api'>__Pyx_PyObject_to_MemoryviewSlice_ds_double</span>(__pyx_v_inarr);\n", | |
| " if (unlikely(!__pyx_t_5.memview)) __PYX_ERR(0, 7, __pyx_L1_error)\n", | |
| " __pyx_v_inarr_memview = __pyx_t_5;\n", | |
| " __pyx_t_5.memview = NULL;\n", | |
| " __pyx_t_5.data = NULL;\n", | |
| "</pre><pre class=\"cython line score-0\"> <span class=\"\">08</span>: </pre>\n", | |
| "<pre class=\"cython line score-28\" onclick='toggleDiv(this)'>+<span class=\"\">09</span>: <span class=\"n\">c_fast_addmul_opt</span><span class=\"p\">(</span><span class=\"n\">inarr_memview</span><span class=\"p\">,</span> <span class=\"n\">outarr</span><span class=\"p\">,</span> <span class=\"nb\">len</span><span class=\"p\">(</span><span class=\"n\">inarr</span><span class=\"p\">),</span> <span class=\"n\">toadd</span><span class=\"p\">,</span> <span class=\"n\">tomul</span><span class=\"p\">)</span></pre>\n", | |
| "<pre class='cython code score-28 '> __pyx_t_5 = <span class='pyx_c_api'>__Pyx_PyObject_to_MemoryviewSlice_ds_double</span>(__pyx_v_outarr);\n", | |
| " if (unlikely(!__pyx_t_5.memview)) __PYX_ERR(0, 9, __pyx_L1_error)\n", | |
| " __pyx_t_6 = <span class='py_c_api'>PyObject_Length</span>(__pyx_v_inarr); if (unlikely(__pyx_t_6 == -1)) __PYX_ERR(0, 9, __pyx_L1_error)\n", | |
| " __pyx_t_7 = __pyx_<span class='py_c_api'>PyFloat_AsDouble</span>(__pyx_v_toadd); if (unlikely((__pyx_t_7 == (double)-1) && <span class='py_c_api'>PyErr_Occurred</span>())) __PYX_ERR(0, 9, __pyx_L1_error)\n", | |
| " __pyx_t_8 = __pyx_<span class='py_c_api'>PyFloat_AsDouble</span>(__pyx_v_tomul); if (unlikely((__pyx_t_8 == (double)-1) && <span class='py_c_api'>PyErr_Occurred</span>())) __PYX_ERR(0, 9, __pyx_L1_error)\n", | |
| " __pyx_t_1 = __pyx_f_46_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5_c_fast_addmul_opt(__pyx_v_inarr_memview, __pyx_t_5, __pyx_t_6, __pyx_t_7, __pyx_t_8); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 9, __pyx_L1_error)\n", | |
| " <span class='refnanny'>__Pyx_GOTREF</span>(__pyx_t_1);\n", | |
| " __PYX_XDEC_MEMVIEW(&__pyx_t_5, 1);\n", | |
| " __pyx_t_5.memview = NULL;\n", | |
| " __pyx_t_5.data = NULL;\n", | |
| " <span class='pyx_macro_api'>__Pyx_DECREF</span>(__pyx_t_1); __pyx_t_1 = 0;\n", | |
| "</pre><pre class=\"cython line score-0\"> <span class=\"\">10</span>: </pre>\n", | |
| "<pre class=\"cython line score-2\" onclick='toggleDiv(this)'>+<span class=\"\">11</span>: <span class=\"k\">return</span> <span class=\"n\">outarr</span></pre>\n", | |
| "<pre class='cython code score-2 '> <span class='pyx_macro_api'>__Pyx_XDECREF</span>(__pyx_r);\n", | |
| " <span class='pyx_macro_api'>__Pyx_INCREF</span>(__pyx_v_outarr);\n", | |
| " __pyx_r = __pyx_v_outarr;\n", | |
| " goto __pyx_L0;\n", | |
| "</pre><pre class=\"cython line score-0\"> <span class=\"\">12</span>: </pre>\n", | |
| "<pre class=\"cython line score-0\"> <span class=\"\">13</span>: <span class=\"nd\">@cython</span><span class=\"o\">.</span><span class=\"n\">boundscheck</span><span class=\"p\">(</span><span class=\"bp\">False</span><span class=\"p\">)</span></pre>\n", | |
| "<pre class=\"cython line score-1\" onclick='toggleDiv(this)'>+<span class=\"\">14</span>: <span class=\"k\">cdef</span> <span class=\"nf\">c_fast_addmul_opt</span><span class=\"p\">(</span><span class=\"n\">double</span><span class=\"p\">[:]</span> <span class=\"n\">inarr</span><span class=\"p\">,</span> <span class=\"n\">double</span><span class=\"p\">[:]</span> <span class=\"n\">outarr</span><span class=\"p\">,</span> <span class=\"nb\">int</span> <span class=\"n\">arrlen</span><span class=\"p\">,</span> <span class=\"n\">double</span> <span class=\"n\">toadd</span><span class=\"p\">,</span> <span class=\"n\">double</span> <span class=\"n\">tomul</span><span class=\"p\">):</span></pre>\n", | |
| "<pre class='cython code score-1 '>static PyObject *__pyx_f_46_cython_magic_eb169e70499a2d4a8f6fd3a12b169ce5_c_fast_addmul_opt(__Pyx_memviewslice __pyx_v_inarr, __Pyx_memviewslice __pyx_v_outarr, int __pyx_v_arrlen, double __pyx_v_toadd, double __pyx_v_tomul) {\n", | |
| " int __pyx_v_i;\n", | |
| " PyObject *__pyx_r = NULL;\n", | |
| " <span class='refnanny'>__Pyx_RefNannyDeclarations</span>\n", | |
| " <span class='refnanny'>__Pyx_RefNannySetupContext</span>(\"c_fast_addmul_opt\", 0);\n", | |
| "/* … */\n", | |
| " /* function exit code */\n", | |
| " __pyx_r = Py_None; <span class='pyx_macro_api'>__Pyx_INCREF</span>(Py_None);\n", | |
| " <span class='refnanny'>__Pyx_XGIVEREF</span>(__pyx_r);\n", | |
| " <span class='refnanny'>__Pyx_RefNannyFinishContext</span>();\n", | |
| " return __pyx_r;\n", | |
| "}\n", | |
| "</pre><pre class=\"cython line score-0\"> <span class=\"\">15</span>: <span class=\"k\">cdef</span> <span class=\"kt\">int</span> <span class=\"nf\">i</span></pre>\n", | |
| "<pre class=\"cython line score-0\" onclick='toggleDiv(this)'>+<span class=\"\">16</span>: <span class=\"k\">for</span> <span class=\"n\">i</span> <span class=\"ow\">in</span> <span class=\"nb\">range</span><span class=\"p\">(</span><span class=\"n\">arrlen</span><span class=\"p\">):</span></pre>\n", | |
| "<pre class='cython code score-0 '> __pyx_t_1 = __pyx_v_arrlen;\n", | |
| " for (__pyx_t_2 = 0; __pyx_t_2 < __pyx_t_1; __pyx_t_2+=1) {\n", | |
| " __pyx_v_i = __pyx_t_2;\n", | |
| "</pre><pre class=\"cython line score-0\" onclick='toggleDiv(this)'>+<span class=\"\">17</span>: <span class=\"n\">outarr</span><span class=\"p\">[</span><span class=\"n\">i</span><span class=\"p\">]</span> <span class=\"o\">=</span> <span class=\"n\">inarr</span><span class=\"p\">[</span><span class=\"n\">i</span><span class=\"p\">]</span><span class=\"o\">*</span><span class=\"n\">tomul</span> <span class=\"o\">+</span> <span class=\"n\">toadd</span></pre>\n", | |
| "<pre class='cython code score-0 '> __pyx_t_3 = __pyx_v_i;\n", | |
| " if (__pyx_t_3 < 0) __pyx_t_3 += __pyx_v_inarr.shape[0];\n", | |
| " __pyx_t_4 = __pyx_v_i;\n", | |
| " if (__pyx_t_4 < 0) __pyx_t_4 += __pyx_v_outarr.shape[0];\n", | |
| " *((double *) ( /* dim=0 */ (__pyx_v_outarr.data + __pyx_t_4 * __pyx_v_outarr.strides[0]) )) = (((*((double *) ( /* dim=0 */ (__pyx_v_inarr.data + __pyx_t_3 * __pyx_v_inarr.strides[0]) ))) * __pyx_v_tomul) + __pyx_v_toadd);\n", | |
| " }\n", | |
| "</pre></div></body></html>" | |
| ], | |
| "text/plain": [ | |
| "<IPython.core.display.HTML object>" | |
| ] | |
| }, | |
| "execution_count": 11, | |
| "metadata": {}, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "%%cython -a\n", | |
| "\n", | |
| "import cython\n", | |
| "import numpy as np\n", | |
| "\n", | |
| "def fast_addmul_opt(inarr, tomul, toadd):\n", | |
| " outarr = np.empty_like(inarr)\n", | |
| " cdef double[:] inarr_memview = inarr\n", | |
| " \n", | |
| " c_fast_addmul_opt(inarr_memview, outarr, len(inarr), toadd, tomul)\n", | |
| " \n", | |
| " return outarr\n", | |
| "\n", | |
| "@cython.boundscheck(False)\n", | |
| "cdef c_fast_addmul_opt(double[:] inarr, double[:] outarr, int arrlen, double toadd, double tomul):\n", | |
| " cdef int i\n", | |
| " for i in range(arrlen):\n", | |
| " outarr[i] = inarr[i]*tomul + toadd" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 12, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "name": "stdout", | |
| "output_type": "stream", | |
| "text": [ | |
| "474 µs ± 20.8 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)\n", | |
| "365 µs ± 39.4 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)\n" | |
| ] | |
| } | |
| ], | |
| "source": [ | |
| "n = 500000\n", | |
| "\n", | |
| "arr = np.arange(n, dtype=float)\n", | |
| "\n", | |
| "%timeit fast_addmul(arr, 1.2, 2)\n", | |
| "%timeit fast_addmul_opt(arr, 1.2, 2)\n", | |
| "None" | |
| ] | |
| }, | |
| { | |
| "cell_type": "markdown", | |
| "metadata": {}, | |
| "source": [ | |
| "# cffi in \"inline\" mode " | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 13, | |
| "metadata": { | |
| "collapsed": true | |
| }, | |
| "outputs": [], | |
| "source": [ | |
| "from cffi import FFI" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 14, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "name": "stdout", | |
| "output_type": "stream", | |
| "text": [ | |
| "generating ./_cffi_addmul_mod.c\n", | |
| "(already up-to-date)\n", | |
| "running build_ext\n", | |
| "building '_cffi_addmul_mod' extension\n", | |
| "clang -Wno-unused-result -Wsign-compare -Wunreachable-code -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -I/Users/erik/miniconda3/include -arch x86_64 -I/Users/erik/miniconda3/include/python3.5m -c _cffi_addmul_mod.c -o ./_cffi_addmul_mod.o\n", | |
| "clang -bundle -undefined dynamic_lookup -L/Users/erik/miniconda3/lib -arch x86_64 ./_cffi_addmul_mod.o -L/Users/erik/miniconda3/lib -o ./_cffi_addmul_mod.cpython-35m-darwin.so\n" | |
| ] | |
| }, | |
| { | |
| "data": { | |
| "text/plain": [ | |
| "'/Users/erik/tmp/_cffi_addmul_mod.cpython-35m-darwin.so'" | |
| ] | |
| }, | |
| "execution_count": 14, | |
| "metadata": {}, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "ffibuilder = FFI()\n", | |
| "\n", | |
| "ffibuilder.cdef(\"void cffi_addmul(double *, double *, double add, double mul, int arrlen);\")\n", | |
| "\n", | |
| "ffibuilder.set_source(\"_cffi_addmul_mod\",\n", | |
| "r\"\"\"\n", | |
| "void cffi_addmul(double * a, double * b, double add, double mul, int arrlen) {\n", | |
| " for (int i=0;i<arrlen;i++){\n", | |
| " b[i] = a[i]*mul+add;\n", | |
| " }\n", | |
| "}\n", | |
| "\"\"\")\n", | |
| "\n", | |
| "ffibuilder.compile(verbose=True)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 15, | |
| "metadata": { | |
| "collapsed": true | |
| }, | |
| "outputs": [], | |
| "source": [ | |
| "from _cffi_addmul_mod import ffi, lib" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 16, | |
| "metadata": { | |
| "collapsed": true | |
| }, | |
| "outputs": [], | |
| "source": [ | |
| "def cffi_addmul(inarr, toadd, tomul):\n", | |
| " outarr = np.empty_like(inarr)\n", | |
| " \n", | |
| " inptr = ffi.cast(\"double *\", ffi.from_buffer(inarr))\n", | |
| " outptr = ffi.cast(\"double *\", ffi.from_buffer(outarr))\n", | |
| " \n", | |
| " lib.cffi_addmul(inptr, outptr, toadd, tomul, len(inarr))\n", | |
| " \n", | |
| " return outarr" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 17, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "data": { | |
| "text/plain": [ | |
| "array([ 1.2, 3.2, 5.2, 7.2, 9.2, 11.2, 13.2, 15.2, 17.2, 19.2])" | |
| ] | |
| }, | |
| "execution_count": 17, | |
| "metadata": {}, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "cffi_addmul(np.arange(10, dtype=float), 1.2, 2)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 18, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "name": "stdout", | |
| "output_type": "stream", | |
| "text": [ | |
| "468 µs ± 7.51 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)\n", | |
| "400 µs ± 26.4 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)\n", | |
| "338 µs ± 25 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)\n" | |
| ] | |
| } | |
| ], | |
| "source": [ | |
| "n = 500000\n", | |
| "\n", | |
| "arr = np.arange(n, dtype=float)\n", | |
| "\n", | |
| "%timeit fast_addmul(arr, 1.2, 2)\n", | |
| "%timeit fast_addmul_opt(arr, 1.2, 2)\n", | |
| "%timeit cffi_addmul(arr, 1.2, 2)\n", | |
| "None" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 21, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "name": "stdout", | |
| "output_type": "stream", | |
| "text": [ | |
| "202 ms ± 9.11 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)\n", | |
| "692 µs ± 5.3 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)\n", | |
| "470 µs ± 7.27 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)\n", | |
| "347 µs ± 7.75 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)\n", | |
| "314 µs ± 10.4 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)\n" | |
| ] | |
| } | |
| ], | |
| "source": [ | |
| "n = 500000\n", | |
| "\n", | |
| "arr = np.arange(n, dtype=float)\n", | |
| "\n", | |
| "%timeit py_addmul(arr, 1.2, 2)\n", | |
| "%timeit np_addmul(arr, 1.2, 2)\n", | |
| "%timeit fast_addmul(arr, 1.2, 2)\n", | |
| "%timeit fast_addmul_opt(arr, 1.2, 2)\n", | |
| "%timeit cffi_addmul(arr, 1.2, 2)\n", | |
| "None" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 28, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "data": { | |
| "text/plain": [ | |
| "(array([ 643.31210191, 2.20382166, 1.49681529, 1.10509554, 1. ]),\n", | |
| " array([ 291.90751445, 1. , 0.67919075, 0.50144509,\n", | |
| " 0.45375723]),\n", | |
| " array([ 1. , 0.00342574, 0.00232673, 0.00171782, 0.00155446]))" | |
| ] | |
| }, | |
| "execution_count": 28, | |
| "metadata": {}, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "us = [202000,\n", | |
| " 692,\n", | |
| " 470,\n", | |
| " 347,\n", | |
| " 314]\n", | |
| "us = np.array(us)\n", | |
| "us/us[-1], us/us[1], us/us[0]" | |
| ] | |
| }, | |
| { | |
| "cell_type": "markdown", | |
| "metadata": {}, | |
| "source": [ | |
| "### DOOM! (maybe?)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 20, | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "data": { | |
| "text/plain": [ | |
| "array([ 4.17232506e-08, 1.90195310e+00, 3.15544367e-31,\n", | |
| " 3.25468755e+00, 2.83953730e+31, 5.50014639e+00,\n", | |
| " 3.85365778e+31, 7.50000906e+00, -3.19744223e-15,\n", | |
| " 1.00000019e+01], dtype=float32)" | |
| ] | |
| }, | |
| "execution_count": 20, | |
| "metadata": {}, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "cffi_addmul(np.arange(10, dtype='float32'), 1.2, 2)" | |
| ] | |
| } | |
| ], | |
| "metadata": { | |
| "kernelspec": { | |
| "display_name": "Python [default]", | |
| "language": "python", | |
| "name": "python3" | |
| }, | |
| "language_info": { | |
| "codemirror_mode": { | |
| "name": "ipython", | |
| "version": 3 | |
| }, | |
| "file_extension": ".py", | |
| "mimetype": "text/x-python", | |
| "name": "python", | |
| "nbconvert_exporter": "python", | |
| "pygments_lexer": "ipython3", | |
| "version": "3.5.2" | |
| } | |
| }, | |
| "nbformat": 4, | |
| "nbformat_minor": 2 | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment