kumquat-buildroot/support/testing/tests/package/test_openblas.py
Julien Olivain 3686e16b52 support/testing/tests/package/test_openblas.py: new runtime test
Signed-off-by: Julien Olivain <ju.o@free.fr>
Signed-off-by: Yann E. MORIN <yann.morin.1998@free.fr>
2023-05-08 22:47:31 +02:00

57 lines
2.0 KiB
Python

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)