#!/usr/bin/env python3 # -*- coding: utf-8 -*- # # Copyright 2017 The ChromiumOS Authors # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Script for running llvm validation tests on ChromeOS. This script launches a buildbot to build ChromeOS with the llvm on a particular board; then it finds and downloads the trybot image and the corresponding official image, and runs test for correctness. It then generates a report, emails it to the c-compiler-chrome, as well as copying the result into a directory. """ # Script to test different toolchains against ChromeOS benchmarks. import argparse import datetime import os import sys import time from cros_utils import buildbot_utils from cros_utils import command_executer from cros_utils import logger CROSTC_ROOT = "/usr/local/google/crostc" ROLE_ACCOUNT = "mobiletc-prebuild" TOOLCHAIN_DIR = os.path.dirname(os.path.realpath(__file__)) MAIL_PROGRAM = "~/var/bin/mail-detective" VALIDATION_RESULT_DIR = os.path.join(CROSTC_ROOT, "validation_result") START_DATE = datetime.date(2016, 1, 1) TEST_PER_DAY = 4 DATA_DIR = "/google/data/rw/users/mo/mobiletc-prebuild/waterfall-report-data/" # Information about Rotating Boards # Board Arch Reference Platform Kernel # Board Version # ------------ ------- ------------ ------------- ------- # atlas x86_64 poppy kabylake-y 4.4.* # cave x86_64 glados skylake-y 3.18 # coral x86_64 reef apollo lake 4.4.* # cyan x86_64 strago braswell 3.18 # elm aarch64 oak mediatek-8173 3.18 arm32 userspace # eve x86_64 poppy kabylake-u/r 4.4.* # gale armv7 3.18 # grunt x86_64 grunt stoney ridge 4.14.* # fizz-moblab x86_64 4.4 # kevin aarch64 gru rockchip-3399 4.4.* arm32 userspace # kevin64 aarch64 gru rockchip-3399 4.4.* arm64 userspace # lakitu x86_64 4.4.* # nyan_kitty armv7 nyan tegra 3.10.18 # octopus x86_64 octopus GLK 4.14.* # sentry x86_64 kunimitsu skylake-u 3.18 # tidus x86_64 auron broadwell 3.14 # veyron_mighty armv7 veyron-pinky rockchip-3288 3.14 # whirlwind armv7 3.14 # winky x86_64 rambi baytrail 4.4.* TEST_BOARD = [ "atlas", "cave", "coral", "cyan", "elm", # 'eve', tested by amd64-llvm-next-toolchain builder. "gale", "grunt", "fizz-moblab", # 'kevin', tested by arm64-llvm-next-toolchain builder. "kevin64", "lakitu", "nyan_kitty", "octopus", "sentry", "tidus", # 'veyron_mighty', tested by arm-llvm-next-toolchain builder. "whirlwind", "winky", ] class ToolchainVerifier(object): """Class for the toolchain verifier.""" def __init__(self, board, chromeos_root, weekday, patches, compiler): self._board = board self._chromeos_root = chromeos_root self._base_dir = os.getcwd() self._ce = command_executer.GetCommandExecuter() self._l = logger.GetLogger() self._compiler = compiler self._build = "%s-%s-toolchain-tryjob" % (board, compiler) self._patches = patches.split(",") if patches else [] self._patches_string = "_".join(str(p) for p in self._patches) if not weekday: self._weekday = time.strftime("%a") else: self._weekday = weekday self._reports = os.path.join(VALIDATION_RESULT_DIR, compiler, board) def DoAll(self): """Main function inside ToolchainComparator class. Launch trybot, get image names, create crosperf experiment file, run crosperf, and copy images into seven-day report directories. """ buildbucket_id, _ = buildbot_utils.GetTrybotImage( self._chromeos_root, self._build, self._patches, tryjob_flags=["--hwtest"], asynchronous=True, ) return buildbucket_id def WriteRotatingReportsData(results_dict, date): """Write data for waterfall report.""" fname = "%d-%02d-%02d.builds" % (date.year, date.month, date.day) filename = os.path.join(DATA_DIR, "rotating-builders", fname) with open(filename, "w", encoding="utf-8") as out_file: for board in results_dict.keys(): buildbucket_id = results_dict[board] out_file.write("%s,%s\n" % (buildbucket_id, board)) def Main(argv): """The main function.""" # Common initializations command_executer.InitCommandExecuter() parser = argparse.ArgumentParser() parser.add_argument( "--chromeos_root", dest="chromeos_root", help="The chromeos root from which to run tests.", ) parser.add_argument( "--weekday", default="", dest="weekday", help="The day of the week for which to run tests.", ) parser.add_argument( "--board", default="", dest="board", help="The board to test." ) parser.add_argument( "--patch", dest="patches", default="", help="The patches to use for the testing, " "seprate the patch numbers with ',' " "for more than one patches.", ) parser.add_argument( "--compiler", dest="compiler", help="Which compiler (llvm, llvm-next or gcc) to use for " "testing.", ) options = parser.parse_args(argv[1:]) if not options.chromeos_root: print("Please specify the ChromeOS root directory.") return 1 if not options.compiler: print( "Please specify which compiler to test (gcc, llvm, or llvm-next)." ) return 1 if options.board: fv = ToolchainVerifier( options.board, options.chromeos_root, options.weekday, options.patches, options.compiler, ) return fv.DoAll() today = datetime.date.today() delta = today - START_DATE days = delta.days start_board = (days * TEST_PER_DAY) % len(TEST_BOARD) results_dict = dict() for i in range(TEST_PER_DAY): try: board = TEST_BOARD[(start_board + i) % len(TEST_BOARD)] fv = ToolchainVerifier( board, options.chromeos_root, options.weekday, options.patches, options.compiler, ) buildbucket_id = fv.DoAll() if buildbucket_id: results_dict[board] = buildbucket_id except SystemExit: logfile = os.path.join( VALIDATION_RESULT_DIR, options.compiler, board ) with open(logfile, "w", encoding="utf-8") as f: f.write("Verifier got an exception, please check the log.\n") WriteRotatingReportsData(results_dict, today) if __name__ == "__main__": retval = Main(sys.argv) sys.exit(retval)