Skip to content
Snippets Groups Projects
Commit 10371533 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Adapt export to new scheduler export signature.

parent c72abf5b
No related branches found
No related tags found
3 merge requests!27v0.2.0,!22v0.4.0,!15Export refactor
import re
import os
from pathlib import Path
import shutil
import numpy as np
from typing import List, Union
from jinja2 import Environment, FileSystemLoader
import aidge_core
from aidge_core.export_utils.data_conversion import aidge2c
from aidge_core.export_utils.code_generation import *
from aidge_core.mem_info import compute_default_mem_info
from aidge_export_cpp.utils import (ROOT, OPERATORS_REGISTRY, supported_operators)
from aidge_export_cpp.utils import ROOT
from aidge_export_cpp.utils.converter import numpy_dtype2ctype
import aidge_export_cpp.operators
from aidge_export_cpp import ExportLibCpp
from aidge_export_cpp.utils.generation import *
# from aidge_export_cpp.memory import *
......@@ -39,10 +33,12 @@ def generate_input_file(export_folder:str,
def export(export_folder_name, graphview, scheduler, mem_wrapping=False):
es = aidge_core.export_utils.ExportScheduler(scheduler)
es.export(export_folder_name,
ExportLibCpp,
memory_manager=compute_default_mem_info)
aidge_core.export_utils.scheduler_export(
scheduler,
export_folder_name,
ExportLibCpp,
memory_manager=compute_default_mem_info
)
# export_folder = Path().absolute() / export_folder_name
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment