Vi presenterar en ny algoritm för att automatiskt schemalägga Halide-program för högpresterande bildbehandling och djupinlärning. Vi förbättrar avsevärt prestandan hos tidigare metoder, som ansågs vara en begränsad delmängd av scheman. Vi definierar en parameterisering av möjliga scheman mycket större än tidigare metoder och använder en variant av strålsökning för att söka över det. Sökningen optimerar körtid som förutses av en kostnadsmodell baserat på en kombination av nya härledda funktioner och maskininlärning. Vi tränar kostnadsmodellen genom att generera och förfalska hundratusentals slumpmässiga program och scheman. Vi visar att detta tillvägagångssätt fungerar effektivt med eller utan autotuning. Det producerar scheman som i genomsnitt är nästan dubbelt så snabba som den befintliga Halide autoscheduler utan autotuning, eller mer än dubbelt så snabbt med, och är den första automatiska schemaläggning algoritm att betydligt överträffa mänskliga experter i genomsnitt. Tillstånd att göra digitala eller papperskopior av hela eller delar av detta arbete för personligt bruk eller klassrum beviljas utan avgift, förutsatt att kopiorna inte görs eller distribueras för vinst eller kommersiella fördelar och att kopiorna är försedda med detta meddelande och den fullständiga hänvisningen på första sidan. Upphovsrätt till delar av detta verk som ägs av andra än upphovsmannen måste hedras. Abstraktering med kredit är tillåten. För att kopiera på annat sätt, eller återpublicera, för att posta på servrar eller för att omfördela till listor, krävs tidigare specifik behörighet och/eller avgift. Begär tillstånd från authoritys@acm.org. Bilda. 1................................................................ Vi genererar scheman för Halide-program med hjälp av trädsökning över utrymmet för scheman (Sec. 3) guidas av en inlärd kostnadsmodell och valfri autotuning (Sec. 4) och av följande skäl: Kostnadsmodellen tränas genom benchmarking tusentals slumpmässigt genererade Halide program och scheman (Sec. 5) När det gäller läkemedel som inte är avsedda att användas tillsammans med andra läkemedel (se avsnitt 4. 5, 4. 5 och 4. Den resulterande koden överträffar avsevärt tidigare arbete och mänskliga experter (Sec. 6 ) och 6 )..................................................................................................
Halide har en nyare autoscheduler REF som använder strålsökning med en inlärd kostnadsmodell för processorer.
Learning to optimize halide with tree search and random programs
{'venue': 'TOGS', 'journal': None, 'mag_field_of_study': ['Mathematics', 'Computer Science']}