|
11 | 11 | import json |
12 | 12 | import logging |
13 | 13 | from datetime import datetime, timezone |
14 | | -from parser import parse_ef # type: ignore |
| 14 | +from edgeflow.parser import parse_ef # type: ignore |
15 | 15 | from pathlib import Path |
16 | 16 | from typing import Any, Dict, List, Optional |
17 | 17 |
|
18 | 18 | # Import core CLI logic |
19 | | -import edgeflowc # type: ignore |
20 | | -from backend.api.services.parser_service import ParserService |
21 | | -from code_generator import CodeGenerator |
| 19 | +import edgeflow.compiler.edgeflowc as edgeflowc # type: ignore |
| 20 | +from edgeflow.backend.api.services.parser_service import ParserService |
| 21 | +from edgeflow.compiler.code_generator import CodeGenerator |
22 | 22 |
|
23 | 23 | # Import EdgeFlow pipeline components |
24 | | -from edgeflow_ast import create_program_from_dict |
25 | | -from edgeflow_ir import FusionPass, IRBuilder, QuantizationPass, SchedulingPass |
26 | | -from explainability_reporter import generate_explainability_report |
27 | | -from fast_compile import fast_compile_config |
| 24 | +from edgeflow.ir.edgeflow_ast import create_program_from_dict |
| 25 | +from edgeflow.ir.edgeflow_ir import ( |
| 26 | + FusionPass, |
| 27 | + IRBuilder, |
| 28 | + QuantizationPass, |
| 29 | + SchedulingPass, |
| 30 | +) |
| 31 | +from edgeflow.reporting.explainability_reporter import generate_explainability_report |
| 32 | +from edgeflow.optimization.fast_compile import fast_compile_config |
28 | 33 | from fastapi import Depends, FastAPI, File, Form, HTTPException, Request, UploadFile |
29 | 34 | from fastapi.middleware.cors import CORSMiddleware |
30 | 35 | from fastapi.responses import JSONResponse |
31 | | -from initial_check import InitialChecker, perform_initial_check |
| 36 | +from edgeflow.analysis.initial_check import InitialChecker, perform_initial_check |
32 | 37 | from pydantic import BaseModel, Field, constr |
33 | 38 | from edgeflow.reporting.reporter import generate_json_report # type: ignore |
34 | 39 |
|
@@ -551,7 +556,11 @@ def fast_compile( |
551 | 556 |
|
552 | 557 | return FastCompileResponse( |
553 | 558 | success=True, |
554 | | - estimated_impact=result.estimated_impact, |
| 559 | + estimated_impact=( |
| 560 | + result.performance_metrics.to_dict() |
| 561 | + if result.performance_metrics |
| 562 | + else None |
| 563 | + ), |
555 | 564 | validation_results={"errors": result.errors, "warnings": result.warnings}, |
556 | 565 | message="Fast compile completed successfully", |
557 | 566 | warnings=result.warnings, |
|
0 commit comments