Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[1/N] Bazel build rules for building TRT-MLIR dialect, running dialect lit tests #298

Draft
wants to merge 28 commits into
base: main
Choose a base branch
from

Conversation

sjain-stanford
Copy link

@sjain-stanford sjain-stanford commented Oct 23, 2024

[Not ready for review yet]

This includes a hermetic python, and CUDA + TensorRT vendored (inside a docker container).

We do have some more work to get to in a ready-for-review state. In particular, I can think of the following:

  • Rebase on the latest main
  • Replace some patches (e.g. in mlir-tensorrt/tensorrt/test/lit.cfg.py and mlir-tensorrt/test/lit.cfg.py) with more principled fixes that don't break CMake build
  • Try to reuse existing requirements instead of adding a new mlir-tensorrt/requirements.txt
  • Reuse existing gpu_tools.py instead of the new mlir-tensorrt/test/gpu_tools.py (this was to mitigate a path issue)
  • If possible maybe get mlir-tensorrt/version.bzl to sync with CMake's version here (so both need not be updated on version bumps)

In follow-on PRs (2/N,...) we (Cruise) will be upstreaming convertors from TCP/TOSA -> TensorRT, and more integrations.

sjain-stanford and others added 28 commits October 23, 2024 11:00
…ests working, still need translate and runner tools
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants