Instead of making a shared_library, let's write a Python package that imports the shared library. It is much easy to test and read, while it adds a little bit of more code.

You can refer to the dummy-nonpure package to see how you can write a Python module that is written in C.

1. I think you can keep the existing implementation, while making a wrapper Python function.

For example, the Python function can be defined such as:

// Python wrapper for simd_avx_add8_sum
static PyObject *
simd_avx_add8_sum(PyObject *self, PyObject *args)
{
  float a0, a1, a2, a3, b0, b1, b2, b3;

  if (!PyArg_ParseTuple(args, "ffffffff", &a0, &a1, &a2, &a3, &b0, &b1, &b2, &b3))
    return NULL;

  float result = simd_avx_add8_sum_impl(a0, a1, a2, a3, b0, b1, b2, b3);

  return Py_BuildValue("f", result);
}

2. The build script can also be moved to setup.py, such as

import os
from setuptools import setup, Extension

module = Extension(
    'test_simd.simd_wrapper',
    sources=[os.path.join('test_simd', 'simd_avx.c')],
    extra_compile_args=['-mavx', '-msimd128'],
)

3. Then it can be tested such as

@run_in_pyodide(packages=["test-simd"])
def test_simd_functions(selenium):
    from test_simd import simd_wrapper
    a = (1.0, 2.0, 3.0, 4.0)
    b = (5.0, 6.0, 7.0, 8.0)

    simd_wrapper.avx_mul(*a, *b)

(The code in this comment is not verified, so some details might be incorrect)

Thank you for the review! I’ll make the changes and commit them.

These are not needed.

• cpython_module type is for unvendored CPython stdlibs
• build script will automatically run for normal Python packages.

Thanks, I’ll update it.

Could you add tag: core as well? It seems like this package is not tested in CI because of a missing tag.

Yes, I’ll add it right away.