aboutsummaryrefslogtreecommitdiffstats
path: root/lib/hipe/regalloc/hipe_regalloc_loop.erl
blob: 5bbb0ba7c147fcd4e8ab06d204c071c6f4e11259 (plain) (blame)
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
%%% -*- erlang-indent-level: 2 -*-
%%%
%%% Licensed under the Apache License, Version 2.0 (the "License");
%%% you may not use this file except in compliance with the License.
%%% You may obtain a copy of the License at
%%%
%%%     http://www.apache.org/licenses/LICENSE-2.0
%%%
%%% Unless required by applicable law or agreed to in writing, software
%%% distributed under the License is distributed on an "AS IS" BASIS,
%%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%%% See the License for the specific language governing permissions and
%%% limitations under the License.
%%%
%%% Common wrapper for graph_coloring and coalescing regallocs.

-module(hipe_regalloc_loop).
-export([ra/7, ra_fp/6]).

%%-define(HIPE_INSTRUMENT_COMPILER, true). %% Turn on instrumentation.
-include("../main/hipe.hrl").

ra(CFG, Liveness0, SpillIndex, Options, RegAllocMod, TargetMod, TargetCtx) ->
  {NewCFG, Liveness, Coloring, _NewSpillIndex} =
    ra_common(CFG, Liveness0, SpillIndex, Options, RegAllocMod, TargetMod,
	      TargetCtx),
  {NewCFG, Liveness, Coloring}.

ra_fp(CFG, Liveness, Options, RegAllocMod, TargetMod, TargetCtx) ->
  ra_common(CFG, Liveness, 0, Options, RegAllocMod, TargetMod, TargetCtx).

ra_common(CFG0, Liveness0, SpillIndex, Options, RegAllocMod, TargetMod,
	  TargetCtx) ->
  ?inc_counter(ra_calls_counter, 1),
  SpillLimit0 = TargetMod:number_of_temporaries(CFG0, TargetCtx),
  {Coloring, _, CFG, Liveness} =
    call_allocator_initial(CFG0, Liveness0, SpillLimit0, SpillIndex, Options,
			   RegAllocMod, TargetMod, TargetCtx),
  %% The first iteration, the hipe_regalloc_prepass may create new temps, these
  %% should not end up above SpillLimit.
  SpillLimit = TargetMod:number_of_temporaries(CFG, TargetCtx),
  alloc(Coloring, CFG, Liveness, SpillLimit, SpillIndex, Options,
	RegAllocMod, TargetMod, TargetCtx).

alloc(Coloring, CFG0, Liveness, SpillLimit, SpillIndex, Options,
      RegAllocMod, TargetMod, TargetCtx) ->
  ?inc_counter(ra_iteration_counter, 1),
  {CFG, DidSpill} = TargetMod:check_and_rewrite(CFG0, Coloring, TargetCtx),
  case DidSpill of
    false -> %% No new temps, we are done.
      ?add_spills(Options, _NewSpillIndex),
      TempMap = hipe_temp_map:cols2tuple(Coloring, TargetMod, TargetCtx),
      {TempMap2, NewSpillIndex2} =
	hipe_spillmin:stackalloc(CFG0, Liveness, [], SpillIndex, Options,
				 TargetMod, TargetCtx, TempMap),
      Coloring2 = 
	hipe_spillmin:mapmerge(hipe_temp_map:to_substlist(TempMap), TempMap2),
      %% case proplists:get_bool(verbose_spills, Options) of
      %%   true ->
      %%     ?msg("Num spill slots used: ~p~n", [NewSpillIndex2-SpillIndex]);
      %%   false ->
      %%     ok
      %% end,
      {CFG, Liveness, Coloring2, NewSpillIndex2};
    _ ->
      %% Since SpillLimit is used as a low-water-mark
      %% the list of temps not to spill is uninteresting.
      {NewColoring, _NewSpillIndex} =
	call_allocator(CFG, Liveness, SpillLimit, SpillIndex, Options,
		       RegAllocMod, TargetMod, TargetCtx),
      alloc(NewColoring, CFG, Liveness, SpillLimit, SpillIndex, Options,
	    RegAllocMod, TargetMod, TargetCtx)
  end.

call_allocator_initial(CFG, Liveness, SpillLimit, SpillIndex, Options,
		       RegAllocMod, TargetMod, TargetCtx) ->
  case proplists:get_bool(ra_prespill, Options) of
    true ->
      hipe_regalloc_prepass:regalloc_initial(
	RegAllocMod, CFG, Liveness, SpillIndex, SpillLimit, TargetMod,
	TargetCtx, Options);
    false ->
      {C, SI} = RegAllocMod:regalloc(CFG, Liveness, SpillIndex, SpillLimit,
				     TargetMod, TargetCtx, Options),
      {C, SI, CFG, Liveness}
  end.

call_allocator(CFG, Liveness, SpillLimit, SpillIndex, Options, RegAllocMod,
	       TargetMod, TargetCtx) ->
  case proplists:get_bool(ra_prespill, Options) of
    true ->
      hipe_regalloc_prepass:regalloc(
	RegAllocMod, CFG, Liveness, SpillIndex, SpillLimit, TargetMod,
	TargetCtx, Options);
    false ->
      RegAllocMod:regalloc(CFG, Liveness, SpillIndex, SpillLimit, TargetMod,
			   TargetCtx, Options)
  end.