| import os |
| |
| import infra.basetest |
| |
| |
| class TestOpenBLAS(infra.basetest.BRTest): |
| config = \ |
| """ |
| BR2_aarch64=y |
| BR2_TOOLCHAIN_EXTERNAL=y |
| BR2_TARGET_GENERIC_GETTY_PORT="ttyAMA0" |
| BR2_LINUX_KERNEL=y |
| BR2_LINUX_KERNEL_CUSTOM_VERSION=y |
| BR2_LINUX_KERNEL_CUSTOM_VERSION_VALUE="6.1.27" |
| BR2_LINUX_KERNEL_USE_CUSTOM_CONFIG=y |
| BR2_LINUX_KERNEL_CUSTOM_CONFIG_FILE="board/qemu/aarch64-virt/linux.config" |
| BR2_LINUX_KERNEL_NEEDS_HOST_OPENSSL=y |
| BR2_TARGET_ROOTFS_CPIO=y |
| BR2_TARGET_ROOTFS_CPIO_GZIP=y |
| # BR2_TARGET_ROOTFS_TAR is not set |
| BR2_PACKAGE_OPENBLAS=y |
| BR2_PACKAGE_OPENBLAS_INSTALL_TESTS=y |
| """ |
| |
| def test_run(self): |
| img = os.path.join(self.builddir, "images", "rootfs.cpio.gz") |
| kern = os.path.join(self.builddir, "images", "Image") |
| self.emulator.boot(arch="aarch64", |
| kernel=kern, |
| kernel_cmdline=["console=ttyAMA0"], |
| options=["-M", "virt", "-cpu", "cortex-a57", "-smp", "2", "-m", "512M", "-initrd", img]) |
| self.emulator.login() |
| |
| test_prefix = "/usr/libexec/openblas/tests" |
| |
| # BLAS data types: |
| blas_data_types = [ |
| "s", # Single precision (32bit) float |
| "d", # Double precision (64bit) double |
| "c", # Single precision (32bit) complex |
| "z" # Double precision (64bit) complex |
| ] |
| |
| # BLAS routine levels: |
| # Level 1: Vector operations, |
| # Level 2: Matrix-Vector operations, |
| # Level 3: Matrix-Matrix operations. |
| for blas_level in range(1, 4): |
| for blas_data_type in blas_data_types: |
| test_name = f"x{blas_data_type}cblat{blas_level}" |
| cmd = test_prefix + "/" + test_name |
| |
| if blas_level > 1: |
| cmd += f" < {test_prefix}/{blas_data_type}in{blas_level}" |
| |
| self.assertRunOk(cmd, timeout=30) |
