mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-07-12 08:22:05 +08:00
remove weights fixup with new SNPE (#24254)
* remove weights fixup with new SNPE * Update ref Co-authored-by: Comma Device <device@comma.ai> Co-authored-by: Harald Schafer <harald.the.engineer@gmail.com>
This commit is contained in:
@@ -65,22 +65,14 @@ common_model = lenv.Object(common_src)
|
||||
if use_thneed and arch == "larch64":
|
||||
fn = File("models/supercombo").abspath
|
||||
compiler = lenv.Program('thneed/compile', ["thneed/compile.cc"]+common_model, LIBS=libs)
|
||||
cmd = f"cd {Dir('.').abspath} && {compiler[0].abspath} --in {fn}.dlc --out {fn}_badweights.thneed --binary --optimize"
|
||||
cmd = f"cd {Dir('.').abspath} && {compiler[0].abspath} --in {fn}.dlc --out {fn}.thneed --binary --optimize"
|
||||
|
||||
lib_paths = ':'.join(Dir(p).abspath for p in lenv["LIBPATH"])
|
||||
kernel_path = os.path.join(Dir('.').abspath, "thneed", "kernels")
|
||||
cenv = Environment(ENV={'LD_LIBRARY_PATH': f"{lib_paths}:{lenv['ENV']['LD_LIBRARY_PATH']}", 'KERNEL_PATH': kernel_path})
|
||||
|
||||
kernels = [os.path.join(kernel_path, x) for x in os.listdir(kernel_path) if x.endswith(".cl")]
|
||||
cenv.Command(fn + "_badweights.thneed", [fn + ".dlc", kernels, compiler], cmd)
|
||||
|
||||
from selfdrive.modeld.thneed.weights_fixup import weights_fixup
|
||||
def weights_fixup_action(target, source, env):
|
||||
weights_fixup(target[0].abspath, source[0].abspath, source[1].abspath)
|
||||
|
||||
env = Environment(BUILDERS = {'WeightFixup' : Builder(action = weights_fixup_action)})
|
||||
env.WeightFixup(target=fn + ".thneed", source=[fn+"_badweights.thneed", fn+".dlc"])
|
||||
|
||||
cenv.Command(fn + ".thneed", [fn + ".dlc", kernels, compiler], cmd)
|
||||
|
||||
lenv.Program('_dmonitoringmodeld', [
|
||||
"dmonitoringmodeld.cc",
|
||||
|
||||
@@ -1 +1 @@
|
||||
f74ab97371be93fdc28333e5ea12bbb78c3a32d0
|
||||
512a9d4596c8faba304d6f7ded2ce77837357b65
|
||||
|
||||
Reference in New Issue
Block a user