import os
import numpy as np

def make_ext(modname, pyxfilename):
  from distutils.extension import Extension
  include_dir = os.path.dirname(pyxfilename)
  return Extension(name=modname,
                   include_dirs=[include_dir, np.get_include()],
                   sources=[pyxfilename],
                   language='c++')