mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-24 06:18:01 -05:00
783c120a8c
* rawcpu * add should work when we respect shapetracker * now that's true * still have to handle shapetracker * copyin * Fix mypy
11 lines
330 B
Plaintext
11 lines
330 B
Plaintext
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++')
|