本文介绍如何使用Python脚本自动化配置Cygwin环境下的Intel MKL库,并创建完整的C++开发环境。通过自动化脚本,我们可以大大简化MKL库的安装和配置过程,提高开发效率。配置完成后,开发者可以直接使用VS Code进行代码编辑、编译和调试,享受完整的开发体验。

概述

Intel Math Kernel Library (MKL) 是一套高度优化的数学库,专门针对Intel处理器进行了优化。在Cygwin环境中配置MKL可以让我们在Windows平台上获得接近Linux环境的开发体验,同时利用MKL的高性能数学计算能力。

Python自动化配置脚本

以下Python脚本将自动完成以下任务:

  1. 下载并安装MKL库
  2. 配置环境变量
  3. 创建示例C++代码文件
  4. 生成Makefile
  5. 配置VS Code调试环境
import os
import sys
import subprocess
import urllib.request
import shutil

def check_cygwin():
    """检查Cygwin环境"""
    if not shutil.which('cygcheck'):
        print("错误: 此脚本需要在Cygwin环境中运行")
        sys.exit(1)
        
    # 检查必要工具
    required_tools = ['wget', 'gcc', 'g++', 'make', 'gdb']
    missing_tools = []
    
    for tool in required_tools:
        if not shutil.which(tool):
            missing_tools.append(tool)
            
    if missing_tools:
        print("错误: 缺少必要的工具: " + ", ".join(missing_tools))
        print("请通过Cygwin安装器安装这些工具")
        sys.exit(1)

def download_mkl():
    """下载MKL安装包"""
    mkl_url = "https://registrationcenter-download.intel.com/akdlm/IRC_NAS/adb8a02c-4ee7-4882-97d6-a524150da358/l_onemkl_p_2023.2.0.49497_offline.sh"
    mkl_script = "l_onemkl_p_2023.2.0.49497_offline.sh"
    
    if not os.path.exists(mkl_script):
        print("下载MKL安装包...")
        try:
            urllib.request.urlretrieve(mkl_url, mkl_script)
            print("下载完成")
        except Exception as e:
            print(f"下载失败: {e}")
            sys.exit(1)
    else:
        print("MKL安装包已存在,跳过下载")
    
    return mkl_script

def install_mkl(mkl_script):
    """安装MKL"""
    print("安装MKL...")
    
    # 使脚本可执行
    os.chmod(mkl_script, 0o755)
    
    # 使用sudo运行安装脚本
    try:
        result = subprocess.run(['sudo', 'sh', './' + mkl_script, '--silent', '--accept-eula'], 
                              check=True, text=True, capture_output=True)
        print("MKL安装完成")
    except subprocess.CalledProcessError as e:
        print(f"MKL安装失败: {e}")
        print(f"错误输出: {e.stderr}")
        sys.exit(1)

def setup_environment():
    """配置环境变量"""
    bashrc_path = os.path.expanduser("~/.bashrc")
    mkl_lib_path = "/opt/intel/oneapi/mkl/latest/lib"
    
    # 检查是否已配置
    with open(bashrc_path, 'r') as f:
        if "LD_LIBRARY_PATH" in f.read() and "mkl" in f.read():
            print("环境变量已配置,跳过")
            return
    
    # 添加环境变量配置
    with open(bashrc_path, 'a') as f:
        f.write("\n# MKL configuration\n")
        f.write(f"export LD_LIBRARY_PATH=\"{mkl_lib_path}:$LD_LIBRARY_PATH\"\n")
        f.write(f"export LIBRARY_PATH=\"{mkl_lib_path}:$LIBRARY_PATH\"\n")
        f.write("source /opt/intel/oneapi/setvars.sh > /dev/null\n")
    
    print("环境变量配置完成")
    
    # 立即生效
    os.environ['LD_LIBRARY_PATH'] = f"{mkl_lib_path}:{os.environ.get('LD_LIBRARY_PATH', '')}"
    os.environ['LIBRARY_PATH'] = f"{mkl_lib_path}:{os.environ.get('LIBRARY_PATH', '')}"

def create_example_project(project_path):
    """创建示例项目"""
    os.makedirs(project_path, exist_ok=True)
    
    # 创建MKL示例代码
    example_code = """/*******************************************************************************
*  Copyright (C) 2009-2015 Intel Corporation. All Rights Reserved.
*  The information and material ("Material") provided below is owned by Intel
*  Corporation or its suppliers or licensors, and title to such Material remains
*  with Intel Corporation or its suppliers or licensors. The Material contains
*  proprietary information of Intel or its suppliers and licensors. The Material
*  is protected by worldwide copyright laws and treaty provisions. No part of
*  the Material may be copied, reproduced, published, uploaded, posted,
*  transmitted, or distributed in any way without Intel's prior express written
*  permission. No license under any patent, copyright or other intellectual
*  property rights in the Material is granted to or conferred upon you, either
*  expressly, by implication, inducement, estoppel or otherwise. Any license
*  under such intellectual property rights must be express and approved by Intel
*  in writing.
*
********************************************************************************
*/
/*
   CGESV Example.
   ==============
 
   The program computes the solution to the system of linear
   equations with a square matrix A and multiple
   right-hand sides B, where A is the coefficient matrix:
 
   (  1.23, -5.50) (  7.91, -5.38) ( -9.80, -4.86) ( -7.32,  7.57)
   ( -2.14, -1.12) ( -9.92, -0.79) ( -9.18, -1.12) (  1.37,  0.43)
   ( -4.30, -7.10) ( -6.47,  2.52) ( -6.51, -2.67) ( -5.86,  7.38)
   (  1.27,  7.29) (  8.90,  6.92) ( -8.82,  1.25) (  5.41,  5.37)

   and B is the right-hand side matrix:
 
   (  8.33, -7.32) ( -6.11, -3.81)
   ( -6.18, -4.80) (  0.14, -7.71)
   ( -5.71, -2.80) (  1.41,  3.40)
   ( -1.60,  3.08) (  8.54, -4.05)
 
   Description.
   ============
 
   The routine solves for X the system of linear equations A*X = B,
   where A is an n-by-n matrix, the columns of matrix B are individual
   right-hand sides, and the columns of X are the corresponding
   solutions.

   The LU decomposition with partial pivoting and row interchanges is
   used to factor A as A = P*L*U, where P is a permutation matrix, L
   is unit lower triangular, and U is upper triangular. The factored
   form of A is then used to solve the system of equations A*X = B.

   Example Program Results.
   ========================
 
 CGESV Example Program Results

 Solution
 ( -1.09, -0.18) (  1.28,  1.21)
 (  0.97,  0.52) ( -0.22, -0.97)
 ( -0.20,  0.19) (  0.53,  1.36)
 ( -0.59,  0.92) (  2.22, -1.00)

 Details of LU factorization
 ( -4.30, -7.10) ( -6.47,  2.52) ( -6.51, -2.67) ( -5.86,  7.38)
 (  0.49,  0.47) ( 12.26, -3.57) ( -7.87, -0.49) ( -0.98,  6.71)
 (  0.25, -0.15) ( -0.60, -0.37) (-11.70, -4.64) ( -1.35,  1.38)
 ( -0.83, -0.32) (  0.05,  0.58) (  0.93, -0.50) (  2.66,  7.86)

 Pivot indices
      3      3      3      4
*/
#include <stdlib.h>
#include <stdio.h>

/* Complex datatype */
struct _fcomplex { float re, im; };
typedef struct _fcomplex fcomplex;

/* CGESV prototype */
extern void cgesv( int* n, int* nrhs, fcomplex* a, int* lda, int* ipiv,
                fcomplex* b, int* ldb, int* info );
/* Auxiliary routines prototypes */
extern void print_matrix( char* desc, int m, int n, fcomplex* a, int lda );
extern void print_int_vector( char* desc, int n, int* a );

/* Parameters */
#define N 4
#define NRHS 2
#define LDA N
#define LDB N

/* Main program */
int main() {
        /* Locals */
        int n = N, nrhs = NRHS, lda = LDA, ldb = LDB, info;
        /* Local arrays */
        int ipiv[N];
        fcomplex a[LDA*N] = {
           { 1.23f, -5.50f}, {-2.14f, -1.12f}, {-4.30f, -7.10f}, { 1.27f,  7.29f},
           { 7.91f, -5.38f}, {-9.92f, -0.79f}, {-6.47f,  2.52f}, { 8.90f,  6.92f},
           {-9.80f, -4.86f}, {-9.18f, -1.12f}, {-6.51f, -2.67f}, {-8.82f,  1.25f},
           {-7.32f,  7.57f}, { 1.37f,  0.43f}, {-5.86f,  7.38f}, { 5.41f,  5.37f}
        };
        fcomplex b[LDB*NRHS] = {
           { 8.33f, -7.32f}, {-6.18f, -4.80f}, {-5.71f, -2.80f}, {-1.60f,  3.08f},
           {-6.11f, -3.81f}, { 0.14f, -7.71f}, { 1.41f,  3.40f}, { 8.54f, -4.05f}
        };
        /* Executable statements */
        printf( " CGESV Example Program Results\n" );
        /* Solve the equations A*X = B */
        cgesv( &n, &nrhs, a, &lda, ipiv, b, &ldb, &info );
        /* Check for the exact singularity */
        if( info > 0 ) {
                printf( "The diagonal element of the triangular factor of A,\n" );
                printf( "U(%i,%i) is zero, so that A is singular;\n", info, info );
                printf( "the solution could not be computed.\n" );
                exit( 1 );
        }
        /* Print solution */
        print_matrix( "Solution", n, nrhs, b, ldb );
        /* Print details of LU factorization */
        print_matrix( "Details of LU factorization", n, n, a, lda );
        /* Print pivot indices */
        print_int_vector( "Pivot indices", n, ipiv );
        exit( 0 );
} /* End of CGESV Example */

/* Auxiliary routine: printing a matrix */
void print_matrix( char* desc, int m, int n, fcomplex* a, int lda ) {
        int i, j;
        printf( "\n %s\n", desc );
        for( i = 0; i < m; i++ ) {
                for( j = 0; j < n; j++ )
                        printf( " (%6.2f,%6.2f)", a[i+j*lda].re, a[i+j*lda].im );
                printf( "\n" );
        }
}

/* Auxiliary routine: printing a vector of integers */
void print_int_vector( char* desc, int n, int* a ) {
        int j;
        printf( "\n %s\n", desc );
        for( j = 0; j < n; j++ ) printf( " %6i", a[j] );
        printf( "\n" );
}
"""
    
    with open(os.path.join(project_path, "main.c"), "w") as f:
        f.write(example_code)
    
    # 创建Makefile
    makefile_content = """TARGET = test

CFLAGS = -std=c99

# -I 指定头文件路径
CFLAGS += -I/opt/intel/oneapi/mkl/latest/include
# -L 指定运行库路径和编译参数
LDFLAGS = -L/opt/intel/oneapi/mkl/latest/lib -lmkl_rt 

#LDFLAGS += -lm -lpthread

test: main.o
	gcc -o test main.o  $(LDFLAGS)  

main.o: main.c
	gcc -c main.c  $(CFLAGS)

clean:
	rm -f *.o test
"""
    
    with open(os.path.join(project_path, "Makefile"), "w") as f:
        f.write(makefile_content)
    
    # 创建编译脚本
    compile_script = """#!/bin/bash
echo "编译MKL示例程序..."
make
if [ $? -eq 0 ]; then
    echo "编译成功,运行程序..."
    ./test
else
    echo "编译失败"
fi
"""
    
    with open(os.path.join(project_path, "compile_run.sh"), "w") as f:
        f.write(compile_script)
    
    os.chmod(os.path.join(project_path, "compile_run.sh"), 0o755)
    
    print(f"示例项目已创建在: {project_path}")

def setup_vscode_config(project_path):
    """配置VS Code调试环境"""
    vscode_dir = os.path.join(project_path, ".vscode")
    os.makedirs(vscode_dir, exist_ok=True)
    
    # 创建launch.json
    launch_json = {
        "version": "0.2.0",
        "configurations": [
            {
                "name": "Cygwin GDB Debug",
                "type": "cppdbg",
                "request": "launch",
                "program": "${workspaceFolder}/test",
                "args": [],
                "stopAtEntry": false,
                "cwd": "${workspaceFolder}",
                "environment": [],
                "externalConsole": False,
                "MIMode": "gdb",
                "miDebuggerPath": "/usr/bin/gdb",
                "setupCommands": [
                    {
                        "description": "为 gdb 启用整齐打印",
                        "text": "-enable-pretty-printing",
                        "ignoreFailures": True
                    }
                ],
                "preLaunchTask": "build-with-mkl"
            }
        ]
    }
    
    with open(os.path.join(vscode_dir, "launch.json"), "w") as f:
        import json
        json.dump(launch_json, f, indent=4)
    
    # 创建tasks.json
    tasks_json = {
        "version": "2.0.0",
        "tasks": [
            {
                "label": "build-with-mkl",
                "type": "shell",
                "command": "make",
                "args": [],
                "group": {
                    "kind": "build",
                    "isDefault": True
                },
                "problemMatcher": [
                    "$gcc"
                ]
            }
        ]
    }
    
    with open(os.path.join(vscode_dir, "tasks.json"), "w") as f:
        import json
        json.dump(tasks_json, f, indent=4)
    
    # 创建c_cpp_properties.json
    cpp_properties = {
        "configurations": [
            {
                "name": "Cygwin",
                "includePath": [
                    "/opt/intel/oneapi/mkl/latest/include",
                    "${workspaceFolder}/**"
                ],
                "defines": [],
                "compilerPath": "/usr/bin/gcc",
                "cStandard": "c11",
                "cppStandard": "c++17",
                "intelliSenseMode": "gcc-x64"
            }
        ],
        "version": 4
    }
    
    with open(os.path.join(vscode_dir, "c_cpp_properties.json"), "w") as f:
        import json
        json.dump(cpp_properties, f, indent=4)
    
    print("VS Code配置已创建")

def main():
    """主函数"""
    print("开始配置Cygwin MKL开发环境")
    
    # 检查Cygwin环境
    check_cygwin()
    
    # 下载MKL
    mkl_script = download_mkl()
    
    # 安装MKL
    install_mkl(mkl_script)
    
    # 配置环境变量
    setup_environment()
    
    # 创建示例项目
    project_path = os.path.join(os.getcwd(), "mkl_example")
    create_example_project(project_path)
    
    # 配置VS Code
    setup_vscode_config(project_path)
    
    print("\n配置完成!")
    print(f"示例项目位于: {project_path}")
    print("使用说明:")
    print("1. cd mkl_example")
    print("2. 运行 ./compile_run.sh 编译并运行示例")
    print("3. 或在VS Code中打开项目文件夹进行调试")

if __name__ == "__main__":
    main()

使用说明

  1. 将上述代码保存为 setup_mkl.py
  2. 在Cygwin终端中运行:
    python setup_mkl.py
    
  3. 按照提示完成安装过程(需要sudo权限)
  4. 安装完成后,进入示例项目目录:
    cd mkl_example
    
  5. 运行编译脚本:
    ./compile_run.sh
    

项目结构

配置完成后,项目目录结构如下:

mkl_example/
├── main.c                 # MKL示例程序
├── Makefile               # 编译配置
├── compile_run.sh         # 一键编译运行脚本
└── .vscode/              # VS Code配置
    ├── launch.json        # 调试配置
    ├── tasks.json         # 任务配置
    └── c_cpp_properties.json # C/C++属性配置

VS Code调试配置

通过上述脚本配置的VS Code环境支持以下功能:

  1. 一键编译:通过Tasks配置,可以使用Ctrl+Shift+B快速编译项目
  2. 调试支持:配置了GDB调试器,可以设置断点、查看变量等
  3. 智能提示:配置了MKL头文件路径,提供代码补全和智能提示功能
Logo

开源鸿蒙跨平台开发社区汇聚开发者与厂商,共建“一次开发,多端部署”的开源生态,致力于降低跨端开发门槛,推动万物智联创新。

更多推荐