nschloe

import numpy as np

import perfplot


def dot(a, b):
    a_shape = a.shape
    b_shape = b.shape
    b = b.reshape(b.shape[0], -1)
    return np.dot(a, b).reshape(a_shape[:-1] + b_shape[1:])

nschloe

import numpy as np
import scipy.optimize

np.random.seed(0)
V_orig = np.random.rand(50, 3)


def func(x):
    Vx = V_orig @ x
    return np.dot(Vx, Vx)

nschloe

import numpy as np
import perfplot


def dot(xy):
    x, y = xy
    return np.dot(x, y)


def einsum(xy):

nschloe

import perfplot
import numpy as np
from scipy.sparse import spdiags


def setup_dense(n):
    A = np.random.rand(n, n) + 1j * np.random.rand(n, n)
    AH = A.T.conj()
    x = np.random.rand(n) + 1j * np.random.rand(n)
    return A, AH, x

nschloe

import math
import numpy as np
import perfplot
import random


def plain_loop(n):
    inside = 0
    random.seed()
    for _ in range(n):

nschloe

import perfplot
import numpy as np


def div(ab):
    a, b = ab
    return a / b


def mul(ab):

nschloe

import numpy as np
import perfplot
from scipy.linalg.blas import daxpy

a = 1.3


def np_axpy(data):
    x, y = data
    return a * x + y

nschloe

import numpy as np
import perfplot
from scipy.linalg.blas import ddot


def np_dot(data):
    x, y = data
    return np.dot(x, y)

nschloe

import numpy as np
import perfplot


def mult(x):
    return x * (1.0 / np.pi)


def div(x):
    return x / np.pi

nschloe

import perfplot
import numpy as np
import npx


A = np.array([[0.0, 1.0, 0.0], [125 / 29, -125 / 29, 0.0], [0.0, 50 / 29, -50 / 29]])


def manual(data):
    fx, fy, fz = data