Brendan Jackman | e81fdcb | 2017-01-04 17:10:29 +0000 | [diff] [blame] | 1 | # Copyright 2015-2017 ARM Limited |
Javi Merino | 034e7cc | 2015-04-22 18:39:21 +0100 | [diff] [blame] | 2 | # |
Javi Merino | aace7c0 | 2015-08-10 14:10:47 +0100 | [diff] [blame] | 3 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | # you may not use this file except in compliance with the License. |
| 5 | # You may obtain a copy of the License at |
| 6 | # |
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | # |
| 9 | # Unless required by applicable law or agreed to in writing, software |
| 10 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | # See the License for the specific language governing permissions and |
| 13 | # limitations under the License. |
| 14 | # |
| 15 | |
Javi Merino | e4c1d45 | 2014-04-01 17:00:47 +0100 | [diff] [blame] | 16 | |
Javi Merino | 86f3bb9 | 2014-08-11 17:07:35 +0100 | [diff] [blame] | 17 | import matplotlib |
Javi Merino | fff3367 | 2014-06-18 16:56:34 +0100 | [diff] [blame] | 18 | import pandas as pd |
| 19 | |
Javi Merino | 076277d | 2014-07-02 18:48:18 +0100 | [diff] [blame] | 20 | from test_thermal import BaseTestThermal |
Javi Merino | 435457c | 2015-08-10 15:59:10 +0100 | [diff] [blame] | 21 | import trappy |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 22 | import cpu_power |
Javi Merino | e4c1d45 | 2014-04-01 17:00:47 +0100 | [diff] [blame] | 23 | |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 24 | class TestCpuPower(BaseTestThermal): |
Javi Merino | 34f9da1 | 2014-06-19 15:48:16 +0100 | [diff] [blame] | 25 | def __init__(self, *args, **kwargs): |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 26 | super(TestCpuPower, self).__init__(*args, **kwargs) |
Javi Merino | d721b27 | 2015-04-20 15:37:53 +0100 | [diff] [blame] | 27 | self.map_label = {"00000000,00000039": "A53", "00000000,00000006": "A57"} |
Javi Merino | 34f9da1 | 2014-06-19 15:48:16 +0100 | [diff] [blame] | 28 | |
Javi Merino | fff3367 | 2014-06-18 16:56:34 +0100 | [diff] [blame] | 29 | def test_pivot_with_labels(self): |
| 30 | """Test pivot_with_labels()""" |
Javi Merino | c68e379 | 2014-07-02 18:14:33 +0100 | [diff] [blame] | 31 | map_label = {"000000f0": "A15", "0000000f": "A7"} |
Javi Merino | fff3367 | 2014-06-18 16:56:34 +0100 | [diff] [blame] | 32 | dfr_in = pd.DataFrame({'cpus': ["000000f0", "0000000f", "000000f0", "0000000f"], |
| 33 | 'freq': [1, 3, 2, 6]}) |
Javi Merino | fff3367 | 2014-06-18 16:56:34 +0100 | [diff] [blame] | 34 | |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 35 | dfr_out = cpu_power.pivot_with_labels(dfr_in, "freq", "cpus", map_label) |
Javi Merino | fff3367 | 2014-06-18 16:56:34 +0100 | [diff] [blame] | 36 | |
| 37 | self.assertEquals(dfr_out["A15"].iloc[0], 1) |
| 38 | self.assertEquals(dfr_out["A15"].iloc[1], 1) |
| 39 | self.assertEquals(dfr_out["A15"].iloc[2], 2) |
| 40 | self.assertEquals(dfr_out["A7"].iloc[1], 3) |
| 41 | |
Javi Merino | 221d44a | 2015-06-17 17:53:34 +0100 | [diff] [blame] | 42 | def test_num_cpus_in_mask(self): |
| 43 | """num_cpus_in_mask() works with the masks we usually use""" |
| 44 | mask = "000000f0" |
| 45 | self.assertEquals(cpu_power.num_cpus_in_mask(mask), 4) |
| 46 | |
| 47 | mask = sorted(self.map_label)[0] |
| 48 | self.assertEquals(cpu_power.num_cpus_in_mask(mask), 2) |
| 49 | |
| 50 | mask = sorted(self.map_label)[1] |
| 51 | self.assertEquals(cpu_power.num_cpus_in_mask(mask), 4) |
| 52 | |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 53 | def test_cpuoutpower_dataframe(self): |
| 54 | """Test that CpuOutPower() creates a proper data_frame""" |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 55 | outp = trappy.FTrace().cpu_out_power |
Javi Merino | e4c1d45 | 2014-04-01 17:00:47 +0100 | [diff] [blame] | 56 | |
Javi Merino | d721b27 | 2015-04-20 15:37:53 +0100 | [diff] [blame] | 57 | self.assertEquals(outp.data_frame["power"].iloc[0], 1344) |
Javi Merino | 92f4d01 | 2014-08-08 17:55:32 +0100 | [diff] [blame] | 58 | self.assertTrue("cdev_state" in outp.data_frame.columns) |
Javi Merino | 62c56dc | 2014-05-07 17:41:12 +0100 | [diff] [blame] | 59 | |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 60 | def test_cpuoutpower_get_all_freqs(self): |
| 61 | """Test CpuOutPower.get_all_freqs()""" |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 62 | dfr = trappy.FTrace().cpu_out_power.get_all_freqs(self.map_label) |
Javi Merino | 73234c1 | 2014-06-18 17:27:16 +0100 | [diff] [blame] | 63 | |
Javi Merino | d721b27 | 2015-04-20 15:37:53 +0100 | [diff] [blame] | 64 | self.assertEquals(dfr["A57"].iloc[0], 1100) |
| 65 | self.assertEquals(dfr["A53"].iloc[1], 850) |
Javi Merino | 73234c1 | 2014-06-18 17:27:16 +0100 | [diff] [blame] | 66 | |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 67 | def test_cpuinpower_get_dataframe(self): |
| 68 | """Test that CpuInPower() creates a proper data_frame()""" |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 69 | inp = trappy.FTrace().cpu_in_power |
Javi Merino | 62c56dc | 2014-05-07 17:41:12 +0100 | [diff] [blame] | 70 | |
Javi Merino | 92f4d01 | 2014-08-08 17:55:32 +0100 | [diff] [blame] | 71 | self.assertTrue("load0" in inp.data_frame.columns) |
Javi Merino | d721b27 | 2015-04-20 15:37:53 +0100 | [diff] [blame] | 72 | self.assertEquals(inp.data_frame["load0"].iloc[0], 24) |
Javi Merino | 8a79fb3 | 2014-05-07 17:56:23 +0100 | [diff] [blame] | 73 | |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 74 | def test_cpuinpower_big_cpumask(self): |
| 75 | """CpuInPower()'s data_frame is not confused by 64-bit cpumasks""" |
Javi Merino | 6649249 | 2015-01-12 15:43:43 +0000 | [diff] [blame] | 76 | in_data = """ kworker/2:2-679 [002] 676.256284: thermal_power_cpu_get: cpus=00000000,0000000f freq=261888 cdev_state=5 power=12 |
| 77 | kworker/2:2-679 [002] 676.276200: thermal_power_cpu_get: cpus=00000000,00000030 freq=261888 cdev_state=5 power=0 |
| 78 | kworker/2:2-679 [002] 676.416202: thermal_power_cpu_get: cpus=00000000,0000000f freq=261888 cdev_state=5 power=0 |
Javi Merino | 35c1ac7 | 2014-07-04 10:29:04 +0100 | [diff] [blame] | 79 | """ |
| 80 | with open("trace.txt", "w") as fout: |
| 81 | fout.write(in_data) |
| 82 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 83 | inp = trappy.FTrace(normalize_time=False).cpu_in_power |
Javi Merino | 92f4d01 | 2014-08-08 17:55:32 +0100 | [diff] [blame] | 84 | self.assertEquals(round(inp.data_frame.index[0], 6), 676.256284) |
Javi Merino | d6284da | 2014-08-14 12:12:13 +0100 | [diff] [blame] | 85 | self.assertEquals(inp.data_frame["cpus"].iloc[1], "00000000,00000030") |
Javi Merino | 35c1ac7 | 2014-07-04 10:29:04 +0100 | [diff] [blame] | 86 | |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 87 | def test_cpuinpower_data_frame_asymmetric_clusters(self): |
| 88 | """Test that CpuInPower()'s data_frame can handle asymmetric clusters |
Javi Merino | 9ce2fb6 | 2014-07-04 20:02:13 +0100 | [diff] [blame] | 89 | |
| 90 | That is 2 cpus in one cluster and 4 in another, like Juno |
| 91 | """ |
Javi Merino | 978ebfb | 2015-02-17 15:30:38 +0000 | [diff] [blame] | 92 | in_data = """ |
| 93 | kworker/2:2-679 [002] 676.256261: thermal_power_cpu_get: cpus=00000000,00000030 freq=1900000 raw_cpu_power=1259 load={74 49} power=451 |
| 94 | kworker/2:2-679 [002] 676.256271: thermal_power_cpu_get: cpus=00000000,0000000f freq=450000 raw_cpu_power=36 load={1 2 1 3} power=9 |
Javi Merino | 9ce2fb6 | 2014-07-04 20:02:13 +0100 | [diff] [blame] | 95 | """ |
| 96 | |
| 97 | with open("trace.txt", "w") as fout: |
| 98 | fout.write(in_data) |
| 99 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 100 | inp = trappy.FTrace(normalize_time=False).cpu_in_power |
Javi Merino | 9ce2fb6 | 2014-07-04 20:02:13 +0100 | [diff] [blame] | 101 | |
Javi Merino | 978ebfb | 2015-02-17 15:30:38 +0000 | [diff] [blame] | 102 | self.assertEquals(inp.data_frame["load0"].iloc[0], 74) |
| 103 | self.assertEquals(inp.data_frame["load1"].iloc[0], 49) |
| 104 | self.assertEquals(inp.data_frame["load2"].iloc[0], 0) |
| 105 | self.assertEquals(inp.data_frame["load3"].iloc[0], 0) |
| 106 | self.assertEquals(inp.data_frame["load0"].iloc[1], 1) |
| 107 | self.assertEquals(inp.data_frame["load1"].iloc[1], 2) |
| 108 | self.assertEquals(inp.data_frame["load2"].iloc[1], 1) |
| 109 | self.assertEquals(inp.data_frame["load3"].iloc[1], 3) |
Javi Merino | 35c1ac7 | 2014-07-04 10:29:04 +0100 | [diff] [blame] | 110 | |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 111 | def test_cpuinpower_get_all_freqs(self): |
| 112 | """Test CpuInPower.get_all_freqs()""" |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 113 | dfr = trappy.FTrace().cpu_in_power.get_all_freqs(self.map_label) |
Javi Merino | 73234c1 | 2014-06-18 17:27:16 +0100 | [diff] [blame] | 114 | |
Javi Merino | d721b27 | 2015-04-20 15:37:53 +0100 | [diff] [blame] | 115 | self.assertEquals(dfr["A57"].iloc[0], 1100) |
| 116 | self.assertEquals(dfr["A53"].iloc[1], 850) |
| 117 | self.assertEquals(dfr["A57"].iloc[5], 1100) |
Javi Merino | 73234c1 | 2014-06-18 17:27:16 +0100 | [diff] [blame] | 118 | |
Javi Merino | 2cfd1e8 | 2015-04-21 17:22:28 +0100 | [diff] [blame] | 119 | def test_cpuinpower_get_load_data(self): |
| 120 | """Test CpuInPower.get_load_data()""" |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 121 | trace = trappy.FTrace() |
| 122 | first_load = trace.cpu_in_power.data_frame["load0"].iloc[0] |
| 123 | load_data = trace.cpu_in_power.get_load_data(self.map_label) |
Javi Merino | 9237a3c | 2014-06-18 12:18:51 +0100 | [diff] [blame] | 124 | |
Javi Merino | d721b27 | 2015-04-20 15:37:53 +0100 | [diff] [blame] | 125 | self.assertEquals(load_data["A57"].iloc[0], 24 + 19) |
| 126 | self.assertEquals(load_data["A53"].iloc[3], 32 + 28 + 46 + 44) |
| 127 | self.assertEquals(load_data["A57"].iloc[0], load_data["A57"].iloc[1]) |
Javi Merino | 7ad9ce0 | 2015-06-16 16:34:10 +0100 | [diff] [blame] | 128 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 129 | self.assertEquals(trace.cpu_in_power.data_frame["load0"].iloc[0], |
Javi Merino | 7ad9ce0 | 2015-06-16 16:34:10 +0100 | [diff] [blame] | 130 | first_load) |
Javi Merino | 86abee1 | 2015-06-16 17:22:06 +0100 | [diff] [blame] | 131 | |
| 132 | def test_cpuinpower_get_normalized_load_data(self): |
| 133 | """Test CpuInPower.get_normalized_load_data()""" |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 134 | trace = trappy.FTrace() |
| 135 | first_load = trace.cpu_in_power.data_frame["load0"].iloc[0] |
| 136 | load_data = trace.cpu_in_power.get_normalized_load_data(self.map_label) |
Javi Merino | 86abee1 | 2015-06-16 17:22:06 +0100 | [diff] [blame] | 137 | |
| 138 | # Ideally the trace should have an event in which the cpus are |
| 139 | # not running at maximum frequency |
| 140 | self.assertEquals(load_data["A57"].iloc[0], |
| 141 | (24. + 19) * 1100000 / (1100000 * 2)) |
| 142 | self.assertEquals(load_data["A53"].iloc[1], |
| 143 | (36. + 49 + 48 + 7) * 850000 / (850000 * 4)) |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 144 | self.assertEquals(trace.cpu_in_power.data_frame["load0"].iloc[0], |
Javi Merino | 86abee1 | 2015-06-16 17:22:06 +0100 | [diff] [blame] | 145 | first_load) |