o i>@s*UdZddlmZddlZddlmZddlmZmZm Z m Z m Z ddl m Z ddlmZmZdd lmZmZdd lmZd Ze jd eZd ed<e jrXeddZneddZdNddZGddde jZGddde Z Gddde Z!Gddde Z"id e d!e d"e d#e d$e d%e d&e d'e d(e d)e d*e d+e d,e d-e!d.dd/d0e!d.dd/d1e!d2d.d/d3e!d4dd/e!d5dd/e!d6d d7e!d.dd/e"e"e"d8d9e"e"d:d9e"d;d9e"ded?<dOdCdDZ$dPdLdMZ%dS)QzRegister allocation tests) annotationsN)Path)CallableListMapping NamedTupleUnion)basic)asmparse)OpcodeRegister)commonchapter_rWRAPPER_SCRIPT librariesz wrapper_osx.szwrapper_linux.si asm.AsmItemreturnboolcs2t|tjrdSd ddtfdd |jDS) z:Is this an instruction that accesses a value on the stack?Foperand asm.OperandrrcSst|tjo |jtjkSN) isinstancer MemorybaserBP)rrc/home/gtoal/gtoal.com/languages/algol60/new-unicode-parser/lang/c/tests4/test_framework/regalloc.pyis_stack szuses_stack..is_stackc3s|]}|VqdSrr).0opr!rr #szuses_stack..N)rrrr)rr Labelanyoperands)rrr$r uses_stacks  r)c@speZdZdZedddZd ddZd!d d Z d"d#ddZd dd$ddZ d dd%ddZ d&d'ddZ dS)( TestRegAllocaTest class for register allocation. We'll generate a test method for each C program in the chapter_20/ directory. Each dynamically generated test calls one of the main test methods defined below: * basic_test: make sure the program behaves correctly but don't inspect the assembly code * no_spills_test: make sure we can allocate every register without spilling * spill_test: the number of spilled pesudos and the number of instructions that access the stack should both be below some upper bound rrcCs |jdS)z'Directory containing extra library coder)test_dirjoinpathselfrrr lib_path2s zTestRegAlloc.lib_pathNonecCs,dd|jdD}|D]}|q dS)zvDelete files produced during this test run (e.g. assembly and object files) Don't delete the wrapper scripts!css<|]}|s|jdvr|jdvr|jtjvr|VqdS)).cz.h) wrapper_osx wrapper_linuxN)is_dirsuffixstemnamer ASSEMBLY_LIBS)r"frrr r%;s   z(TestRegAlloc.tearDown..*N)r+rglobunlink)r. garbage_filesr9rrr tearDown7s   zTestRegAlloc.tearDown program_pathcCs&t|dtg}|||dS)aTest that the compiled program behaves correctly but don't inspect the assembly code. Compile the program, linking against the wrapper script (which defines main) and any extra libraries, then run it and validate the result. Tests that _do_ inspect the assembly code should first call basic_test to make sure the program behaves correctly, then parse the assembly file and perform further validation. Args: program_path: Absolute path to C or assembly file to compile and run r1N)r get_libs with_suffixrlibrary_test_helper)r.r? extra_libsrrr basic_testGs zTestRegAlloc.basic_testtarget target_funstrasm.AssemblyFunctionc Csvz |j|ddWntjy'}z|d|jWYd}~nd}~ww|d}||t ||}|S)aShared logic for register allocation tests that validate assembly code. 1. Compile the file at program_path to assembly 2. Call basic_test to make sure it behaves correctly 3. Parse assembly file and return it The caller can then perform further validation on the parsed assembly code Args: program_path: Absolute path to C file under test target_fun: Name of function to parse/inspect Returns: Parsed assembly code for specified target fun z-s)cc_optzCompilation failed: Nz.s) invoke_compilercheck_returncode subprocessCalledProcessErrorfailstderrrArDr parse_file)r.r?rFeasm_file parsed_asmrrr run_and_parseVs  zTestRegAlloc.run_and_parserFcCs>|j||d}dd|jD}|j|tjd|||dddS)aQTest that we allocated every register in target_fun without spilling. First make sure behavior is correct, then examine parsed assembly to validate that we never access the stack Args: program_path: Absolute path to C file under test target_fun: Name of function to parse/inspect r?rFcSg|]}t|r|qSrr)r"rrrr z/TestRegAlloc.no_spills_test..1Found instructions that use operands on the stackbad_instructions full_progr?msgN)rT instructions assertFalser build_msg)r.r?rFrSr^rrr no_spills_testws zTestRegAlloc.no_spills_testmax_spilled_instructionsintmax_spilled_pseudosc Cs|j||d}dd|jD}|jt||tjd|d|||ddtdd|D}|jt||tjd |d |||ddd S) aoTest for a program with so many conflicts that it spills. First validate the compiled program's behavior, then make sure we don't have more than the expected number of distinct spilled pseudoregisters, or produce more than the expected number of instructions that access the stack Args: program_path: Absolute path to C file under test max_spilled_instructions: maximum number of instructions that access the stack max_spilled_pseudos: maximum number of distinct stack addresses accessed target_fun: Name of function to parse/inspect rVcSs$g|]}t|r|jtjkr|qSr)r)opcoder MOVrYrrr rZs z+TestRegAlloc.spill_test..zShould only need zJ instructions involving spilled pseudo but found {len(spill_instructions)}r]r`cSs,g|]}|jD] }t|tjrt|qqSr)r(rr rrG)r"rr#rrr rZs zAt most zM pseudoregs should have been spilled, looks like {len(spilled_operands)} wereN)rTrbassertLessEquallenrrdset)r.r?rfrhrFrSspill_instructionsspilled_operandsrrr spill_tests@    zTestRegAlloc.spill_testr max_movesc sddd|j||d}dd |jD}fd d |jD}|j|tjd |||d d |jt||tjd|dt||||d d dS)aTest that we perform register coalescing properly. First validate the compiled program's behavior, then make sure we don't have more than the expected number of mov instructions where the source and destination are both registers. Also validate that there are no spills. Args: program_path: Absolute path to C file under test target_fun: Name of function to parse/inspect max_moves: maximum number of mov instructions between registers rrrrcSs\t|r,|jd|jd}}t|tjo+|tjtjfvo+t|tjo+|tjtjfvSdS)zCCheck whether this is a move between registers (other than RBP/RSP)rr F)ris_movr(rr rrSP)rsrcdstrrr is_mov_between_regss   z9TestRegAlloc.coalescing_test..is_mov_between_regsrUcSrWrrXrYrrr rZr[z0TestRegAlloc.coalescing_test..csg|]}|r|qSrrrYrvrr rZs r\r]r`zExpected at most z move instructions but found Nrrrr)rTrbrcrrdrkrl)r.r?rFrqrSr^mov_instructionsrrwr coalescing_tests8    zTestRegAlloc.coalescing_testN)rr)rr0)r?rrr0)rE)r?rrFrGrrH)r?rrFrGrr0) r?rrfrgrhrgrFrGrr0)rEr)r?rrFrGrqrgrr0) __name__ __module__ __qualname____doc__propertyr/r>rDrTrerprzrrrr r*&s    %'Ar*c@seZdZUdZded<dS) NoSpillTestrErGrFN)r{r|r}rF__annotations__rrrr rs rc@s*eZdZUded<ded<dZded<dS) SpillTestrgrhrfrErGrFN)r{r|r}rrFrrrr rs rc@s&eZdZUdZded<dZded<dS) CoalesceTestrErGrFrrgrqN)r{r|r}rFrrqrrrr rs rztrivially_colorable.czuse_all_hardregs.czpreserve_across_fun_call.cztrack_arg_registers.czmany_pseudos_fewer_conflicts.czcmp_no_updates.czcopy_no_interference.czsame_instr_no_interference.czloop.czdbl_trivially_colorable.czfourteen_pseudos_interfere.cztrack_dbl_arg_registers.czstore_pointer_in_register.cz force_spill.c)rfrhzforce_spill_mixed_ints.czrewrite_regression_test.c ztest_spill_metric.c)rhrf)rq) ztest_spill_metric_2.czoptimistic_coloring.czforce_spill_doubles.czbriggs_coalesce.czgeorge_coalesce.czcoalesce_prevents_spill.czbriggs_coalesce_hardreg.czbriggs_dont_coalesce.czgeorge_dont_coalesce.czgeorge_dont_coalesce_2.czno_george_test_for_pseudos.czgeorge_off_by_one.cz9Mapping[str, Union[CoalesceTest, NoSpillTest, SpillTest]]REGALLOC_TESTSprogram no_coalescingCallable[[TestRegAlloc], None]cstj}|durtSdjvr|rd fdd }|St|tr0|d fd d }|St|trA|d fd d }|S|d fd d }|S) z/Generate test method for a single test program.Nwith_coalescingr.r*rr0cs|dSr)rDr-)rrr testmsz make_regalloc_test..testcs|jjddS)NrU)rerFr-)nospilltest_inforrr rus cs|jjjjddS)N)rfrhrF)rprfrhrFr-)rspilltest_inforr r~s  cs|jjjddS)N)rFrq)rzrFrqr-)rtirr rs  )r.r*rr0) rgetr7r make_test_runpartsrrr)rr test_inforr)rrrrr make_regalloc_test]s"  #  rcompileroptions List[str]extra_credit_flagsbasic.ExtraCreditint_onlyr0c Csttdtttd|ttd|ttdd|rdg}nddg}dd |D}|D]'}t||r4q+|td }d |} tt| drIJtt| t||q+dS) aDynamically add test methods and attributes to TestRegAlloc. Args: compiler: absolute path to compiler under test options: extra command-line arguments to pass through to compiler extra_credit_flags: extra credit features to test, represented as a bit vector int_only: the reader skipped Part II; only include tests that rely on Part I language features no_coalescing: the reader hasn't implemented register coalescing yet, so don't test for it r+ccr exit_stageNr all_typescSs$g|]}t|dD]}|q qS)z*.c)TEST_DIRr;)r"subdirprrr rZs$z#configure_tests..test_) setattrr*rr excluded_extra_credit relative_torAgetattrr) rrrrrsubdirs all_testsrkeyr7rrr configure_testss$      rrx)rrrrrr) rrrrrrrrrrrr0)&r~ __future__rrLpathlibrtypingrrrrrrr parserr r parser.asmr rtackyrCHAPTERrr,resolverIS_OSXrr) TestChapterr*rrrrrrrrrr s       p      96