Skip to content

Add tilegym wheel building#41

Merged
arjkesh merged 11 commits intoNVIDIA:mainfrom
arjkesh:wheel_builder
Jan 30, 2026
Merged

Add tilegym wheel building#41
arjkesh merged 11 commits intoNVIDIA:mainfrom
arjkesh:wheel_builder

Conversation

@arjkesh
Copy link
Collaborator

@arjkesh arjkesh commented Jan 29, 2026 •
edited
Loading

Description

Add wheel build for tilegym

CI Configuration

config:
  build: true
  # valid options are "ops" and "benchmark"
  test: ["ops", "benchmark"]

Checklist

  • Code formatted and imports sorted via repo specifications (./format.sh)
  • Documentation updated (if needed)
  • CI configuration reviewed

@copy-pr-bot
Copy link

copy-pr-bot bot commented Jan 29, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@arjkesh
Copy link
Collaborator Author

arjkesh commented Jan 29, 2026

/ok to test 38b50eb

@arjkesh
Copy link
Collaborator Author

arjkesh commented Jan 29, 2026

/ok to test 6df7004

@arjkesh
Copy link
Collaborator Author

arjkesh commented Jan 29, 2026

/ok to test ef64d04

xjmxyt
xjmxyt reviewed Jan 29, 2026
View reviewed changes
modeling/transformers/Dockerfile Outdated
# TileGym Transformers Inference Container
# This is a simplified Dockerfile for running TileGym-accelerated transformer inference
#
# This Dockerfile expects a pre-built wheel in the ./wheel directory
Copy link
Collaborator

@xjmxyt xjmxyt Jan 29, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think if we should have two options: one with whell and one without. And the default option is without wheel. Because currently I assume most users just directly build docker with this Dockerfile but without wheel.

Copy link
Collaborator Author

@arjkesh arjkesh Jan 30, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is a good idea. I reimplemented this as a multistage build (one with a CI target, the other with a general target) so that OSS users can still install tilegym from source by default

@arjkesh
Copy link
Collaborator Author

arjkesh commented Jan 30, 2026

/ok to test cd6ec50

@arjkesh
Copy link
Collaborator Author

arjkesh commented Jan 30, 2026

/ok to test cf893da

@arjkesh
Copy link
Collaborator Author

arjkesh commented Jan 30, 2026

/ok to test 3fb5517

@arjkesh
Copy link
Collaborator Author

arjkesh commented Jan 30, 2026

/ok to test dee3fd4

@arjkesh arjkesh enabled auto-merge (squash) January 30, 2026 17:19
xjmxyt
xjmxyt approved these changes Jan 30, 2026
View reviewed changes
Copy link
Collaborator

@xjmxyt xjmxyt left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@arjkesh arjkesh merged commit 68b114b into NVIDIA:main Jan 30, 2026
18 checks passed
@arjkesh arjkesh deleted the wheel_builder branch January 31, 2026 03:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@xjmxyt xjmxyt xjmxyt approved these changes

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@arjkesh @xjmxyt