test

Author: Changming Sun

.github/actions/build-docker-image/action.yml

@@ -0,0 +1,17 @@
+name: 'Build Docker Image'
+description: 'Builds a Docker image and optionally pushes it to Azure Container Registry.'
+inputs:
+  Dockerfile:
+    description: 'Path to the Dockerfile.'
+    required: true
+  DockerBuildArgs:
+    description: 'Arguments to pass to docker build.'
+    required: false
+    default: ''
+  Repository:
+    description: 'The image repository name.'
+    required: true
+
+runs:
+  using: 'node20'
+  main: 'index.js'

.github/actions/build-docker-image/index.js

@@ -0,0 +1,76 @@
+// index.js (or main.js - see package.json "main" field)
+
+const core = require('@actions/core');
+const exec = require('@actions/exec');
+const github = require('@actions/github');
+
+async function run() {
+  try {
+    // Get inputs
+    const dockerfile = core.getInput('Dockerfile', { required: true });
+    const dockerBuildArgs = core.getInput('DockerBuildArgs');  // Defaults to "" if not provided.
+    const repository = core.getInput('Repository', { required: true });
+
+    const context = dockerfile.substring(0, dockerfile.lastIndexOf('/')) || '.'; // Extract directory of Dockerfile. or .
+    core.info(`Dockerfile context directory: ${context}`);
+
+    // Determine if we should use the cache.  Avoid cache if it's a PR from a fork.
+    const isPullRequest = github.context.eventName === 'pull_request';
+    const isFork = isPullRequest && github.context.payload.pull_request.head.repo.fork;
+    const useCache = !isFork;
+
+    let azLoginRan = false;
+
+    if (useCache) {
+      // Log in to Azure
+      try {
+        await exec.exec('az', ['login', '--identity']);
+        azLoginRan = true;
+        await exec.exec('az', ['acr', 'login', '-n', 'onnxruntimebuildcache']);
+      } catch (error) {
+        core.setFailed(`Azure login or ACR login failed: ${error.message}`);
+        return; // Exit if login fails.  Critical error.
+      }
+    }
+
+
+    // Build arguments for get_docker_image.py
+    let scriptArgs = [
+      'tools/ci_build/get_docker_image.py',
+      '--dockerfile', dockerfile,
+      '--context', context,
+      '--docker-build-args', dockerBuildArgs,
+      '--repository', repository,
+    ];
+
+    if (useCache) {
+      scriptArgs.push('--container-registry', 'onnxruntimebuildcache');
+      scriptArgs.push('--use_imagecache');
+    }
+
+
+    // Execute the Python script.
+    try {
+      await exec.exec('python3', scriptArgs);
+    } catch (error) {
+      core.setFailed(`Docker build failed: ${error.message}`);
+      return; // Exit if docker build failed.
+    }
+
+    // Logout from Azure ACR (if we logged in)
+    if (azLoginRan) {
+      try {
+        await exec.exec('az', ['acr', 'logout', '-n', 'onnxruntimebuildcache']);
+      } catch (error) {
+        core.warning(`Azure ACR logout failed: ${error.message}`); // Warning, not critical.
+      }
+    }
+
+    core.info('Docker build completed successfully.');
+
+  } catch (error) {
+    core.setFailed(error.message);
+  }
+}
+
+run();

.github/workflows/linux_ci.yml

@@ -0,0 +1,36 @@
+name: Linux CI
+
+on:
+  push:
+    branches: [ main, 'rel-*', 'snnn/linux_ci']
+  pull_request:
+    branches: [ main ]
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v3
+
+      - name: Set up Python 3.x
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.x'
+
+      - name: Install dependencies
+        run: python -m pip install -r requirements-dev.txt        
+        
+      - name: Build Docker Image
+        uses: ./.github/actions/build-docker-image  # Path to your action directory
+        with:
+          Dockerfile: ${{ github.workspace }}/tools/ci_build/github/linux/docker/inference/x86_64/default/cpu/Dockerfile
+          DockerBuildArgs: "--build-arg BUILD_UID=$( id -u )  --container-registry onnxruntimebuildcache --repository onnxruntimecpubuildcentos8x64"
+          Repository: 'onnxruntimecpubuildcentos8x64'
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}  # Needed for context