File size: 1,004 Bytes
02c783d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
from typing import List, Optional, Any
from dataclasses import dataclass, field
@dataclass
class ProblemState:
filename: str
label: Optional[str] = None
test_code: Optional[str] = None
instruction: Optional[str] = None
solution: Optional[str] = None
speedup: float = 0.0
@dataclass
class ProblemStateROCm:
instruction: str
label: str
filename: str
opname: str
target_kernel_name: str
test_code: str
solution: Optional[str] = None
pass_call: bool = False
pass_exe: bool = False
speedup: float = 0.0
@dataclass
class tempCode:
code: Optional[str] = None
strategy: Optional[str] = None
reflections: Optional[str] = None
test_stdout: Optional[str] = None
test_stderr: Optional[str] = None
profilig: Optional[str] = None
pass_call: bool = False
pass_exe: bool = False
pass_perf: bool = False
latency: float = 0.0
eff: float = 0.0
llm_metric: float = 0.0
llm_eval: Optional[str] = None
|