JupyterのCell MagicでFortranやC++やCythonを使う

!pip3 install -U fortran-magic

%load_ext fortranmagic

%%fortran
subroutine compute_fortran(x, y, z)
    real, intent(in) :: x(:), y(:)
    real, intent(out) :: z(size(x, 1))

    z = sin(x + y)

end subroutine compute_fortran

compute_fortran([1, 2, 3], [4, 5, 6])

%%fortran --link lapack

subroutine solve(A, b, x, n)
    ! solve the matrix equation A*x=b using LAPACK
    implicit none

    real*8, dimension(n,n), intent(in) :: A
    real*8, dimension(n), intent(in) :: b
    real*8, dimension(n), intent(out) :: x

    integer :: pivot(n), ok

    integer, intent(in) :: n
    x = b

    ! find the solution using the LAPACK routine SGESV
    call DGESV(n, 1, A, n, pivot, x, n, ok)

end subroutine

A = np.array([[1, 2.5], [-3, 4]])
b = np.array([1, 2.5])

solve(A, b)

%load_ext Cython

%%cython
def power_of_two(int x):
    return 2.0 ** x

power_of_two(256) #2の256乗

%%cython --compile-args=-fopenmp --link-args=-fopenmp --force
cimport cython
cimport openmp

import cython.parallel as cp
from cython.parallel import parallel, prange

import numpy as np
cimport numpy as np

ctypedef np.int32_t int32_t

cdef extern from "unistd.h" nogil:
    unsigned int sleep(unsigned int seconds)

@cython.boundscheck(False)
cpdef double[::1] ompfunc(int32_t size, int32_t num_threads):
    cdef int32_t idx
    cdef double[::1] result = np.empty(size, dtype=np.float64)
    with nogil, cp.parallel(num_threads=num_threads):
        for idx in prange(size, schedule='dynamic'):
            result[idx] = cp.threadid()
            sleep(1)
    return result

np.asarray(ompfunc(10, 10))

%timeit ompfunc(10, 1)
%timeit ompfunc(10, 2)
%timeit ompfunc(10, 3)
%timeit ompfunc(10, 4)
%timeit ompfunc(10, 5)
%timeit ompfunc(10, 6)
%timeit ompfunc(10, 7)
%timeit ompfunc(10, 8)
%timeit ompfunc(10, 9)
%timeit ompfunc(10, 10)

%%file fib.hpp

double fib(int n);

%%file fib.pxd

cdef extern from "fib.hpp":
    double fib(int n)

%%file fib.cpp

double fib(int n) {
    double a = 0, b = 1;
    for (int i=0; i<n; i++) {
        double tmp = b;
        b = a;
        a += tmp;
     }
    return a;
}

%%file setup.py
from distutils.core import setup, Extension
from Cython.Build import cythonize

ext = Extension("fib2cpp",
              sources=["fib2cpp.pyx", "fib.cpp"],
              language="c++",)

setup(name = "cython_fibcpp",
      ext_modules = cythonize(ext))

%%file fib2cpp.pyx

cimport fib

def fib(n):
    return fib.fib(n)

! python setup.py build_ext --inplace

import fib2cpp
fib2cpp.fib(100)