Skip to content

Conversation

chentong319
Copy link
Collaborator

@chentong319 chentong319 commented Feb 25, 2025

Previous driver only works with python docker package. This PR adds the support for podman package.
The interface of the docker and podman package is the same for the features used in this driver so far.
Another change in this PR is to pull the image if it does not exist locally.

Three cases tested:
Inside onnx-mlir-dev container:
sess = onnxmlir.InferenceSession("gather-2.mlir", compiler_path="../build-2024/Debug/bin/onnx-mlir", compiler_image_name=None)

outside container:

  1. sess = onnxmlir.InferenceSession("gather-2.mlir", compiler_container_engine="docker")
  2. sess = onnxmlir.InferenceSession("gather-2.mlir", compiler_container_engine="podman")

Signed-off-by: Chen Tong <[email protected]>
Signed-off-by: Chen Tong <[email protected]>
@chentong319 chentong319 merged commit 7da9acb into onnx:main Feb 26, 2025
7 checks passed
@chentong319 chentong319 deleted the pyruntie-3 branch February 26, 2025 14:44
@jenkins-droid
Copy link
Collaborator

Jenkins Linux amd64 Build #16352 [push] Use docker and podman pa... started at 08:45

@jenkins-droid
Copy link
Collaborator

Jenkins Linux s390x Build #16354 [push] Use docker and podman pa... started at 09:45

@jenkins-droid
Copy link
Collaborator

Jenkins Linux ppc64le Build #15335 [push] Use docker and podman pa... started at 09:56

@jenkins-droid
Copy link
Collaborator

Jenkins Linux amd64 Build #16352 [push] Use docker and podman pa... passed after 1 hr 21 min

@jenkins-droid
Copy link
Collaborator

Jenkins Linux s390x Build #16354 [push] Use docker and podman pa... passed after 1 hr 29 min

@jenkins-droid
Copy link
Collaborator

Jenkins Linux ppc64le Build #15335 [push] Use docker and podman pa... passed after 2 hr 21 min

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants