|
|
|
@ -4,7 +4,7 @@ Import('env', 'envCython', 'arch', 'cereal', 'messaging', 'common', 'gpucommon', |
|
|
|
|
lenv = env.Clone() |
|
|
|
|
lenvCython = envCython.Clone() |
|
|
|
|
|
|
|
|
|
libs = [cereal, messaging, visionipc, gpucommon, common, 'capnp', 'zmq', 'kj', 'pthread'] |
|
|
|
|
libs = [cereal, messaging, visionipc, gpucommon, common, 'capnp', 'kj', 'pthread'] |
|
|
|
|
frameworks = [] |
|
|
|
|
|
|
|
|
|
common_src = [ |
|
|
|
@ -69,6 +69,6 @@ if arch == "larch64" or GetOption('pc_thneed'): |
|
|
|
|
|
|
|
|
|
lenv.Command(fn + ".thneed", [fn + ".onnx"] + tinygrad_files, cmd) |
|
|
|
|
|
|
|
|
|
thneed_lib = env.SharedLibrary('thneed', thneed_src, LIBS=[gpucommon, common, 'zmq', 'OpenCL', 'dl']) |
|
|
|
|
thneed_lib = env.SharedLibrary('thneed', thneed_src, LIBS=[gpucommon, common, 'OpenCL', 'dl']) |
|
|
|
|
thneedmodel_lib = env.Library('thneedmodel', ['runners/thneedmodel.cc']) |
|
|
|
|
lenvCython.Program('runners/thneedmodel_pyx.so', 'runners/thneedmodel_pyx.pyx', LIBS=envCython["LIBS"]+[thneedmodel_lib, thneed_lib, gpucommon, common, 'dl', 'zmq', 'OpenCL']) |
|
|
|
|
lenvCython.Program('runners/thneedmodel_pyx.so', 'runners/thneedmodel_pyx.pyx', LIBS=envCython["LIBS"]+[thneedmodel_lib, thneed_lib, gpucommon, common, 'dl', 'OpenCL']) |
|
|
|
|