Skip to content
Navigation Menu
Sign in
Appearance settings
Platform
AI CODE CREATION
GitHub Copilot
Write better code with AI
GitHub Copilot app
Direct agents from issue to merge
MCP Registry
Integrate external tools
DEVELOPER WORKFLOWS
Actions
Automate any workflow
Codespaces
Instant dev environments
Issues
Plan and track work
Code Review
Manage code changes
Code Quality
Enforce quality at merge
APPLICATION SECURITY
GitHub Advanced Security
Find and fix vulnerabilities
Code security
Secure your code as you build
Secret protection
Stop leaks before they start
EXPLORE
Why GitHub
Documentation
Blog
Changelog
Marketplace
View all features
Solutions
BY COMPANY SIZE
Enterprises
Small and medium teams
Startups
Nonprofits
BY USE CASE
App Modernization
DevSecOps
DevOps
CI/CD
View all use cases
BY INDUSTRY
Healthcare
Financial services
Manufacturing
Government
View all industries
View all solutions
Resources
EXPLORE BY TOPIC
AI
Software Development
DevOps
Security
View all topics
EXPLORE BY TYPE
Customer stories
Events & webinars
Ebooks & reports
Business insights
GitHub Skills
SUPPORT & SERVICES
Documentation
Customer support
Community forum
Trust center
Partners
View all resources
Open Source
COMMUNITY
GitHub Sponsors
Fund open source developers
PROGRAMS
Security Lab
Maintainer Community
Accelerator
GitHub Stars
Archive Program
REPOSITORIES
Topics
Trending
Collections
Enterprise
ENTERPRISE SOLUTIONS
Enterprise platform
AI-powered developer platform
AVAILABLE ADD-ONS
GitHub Advanced Security
Enterprise-grade security features
Copilot for Business
Enterprise-grade AI features
Premium Support
Enterprise-grade 24/7 support
Pricing
Type
/
to search
Sign in
Sign up
Appearance settings
You signed in with another tab or window.
Reload
to refresh your session.
You signed out in another tab or window.
Reload
to refresh your session.
You switched accounts on another tab or window.
Reload
to refresh your session.
Dismiss alert
{{ message }}
Uh oh!
There was an error while loading.
Please reload this page
.
microsoft
/
dist-ir
Public
Notifications
You must be signed in to change notification settings
Fork
2
Star
33
Code
Issues
0
Pull requests
7
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Actions
Projects
Security and quality
Insights
Add ONNX exporter
- #26
#26
Draft
Keshav Santhanam (santhnm2)
wants to merge 63 commits into
main
microsoft/dist-ir:main
from
export_to_onnx
microsoft/dist-ir:export_to_onnx
Copy head branch name to clipboard
Conversation
Commits
63
(63)
Checks
Files changed
Draft
Add ONNX exporter
#26
Keshav Santhanam (santhnm2)
wants to merge 63 commits into
main
microsoft/dist-ir:main
from
export_to_onnx
microsoft/dist-ir:export_to_onnx
Copy head branch name to clipboard
Commits
Commits on Apr 19, 2021
Projector: DistIR function -> per-rank function
Siddharth Krishna (siddharth-krishna)
committed
8ac2c50
View commit details
Copy full SHA for 8ac2c50
Browse repository at this point
Fix bug: constructing variadic ops when specifying output_values
Siddharth Krishna (siddharth-krishna)
committed
193dc84
View commit details
Copy full SHA for 193dc84
Browse repository at this point
Fix sequential executor docstring
Siddharth Krishna (siddharth-krishna)
committed
c5bfbd9
View commit details
Copy full SHA for c5bfbd9
Browse repository at this point
Commits on Apr 25, 2021
A distributed PyTorch backend
Siddharth Krishna (siddharth-krishna)
committed
d2f9e0a
View commit details
Copy full SHA for d2f9e0a
Browse repository at this point
Upgrade PyTorch version
Siddharth Krishna (siddharth-krishna)
committed
d9cc787
View commit details
Copy full SHA for d9cc787
Browse repository at this point
Add test: one-weird-trick (matmul version)
Siddharth Krishna (siddharth-krishna)
committed
a2a4cc2
View commit details
Copy full SHA for a2a4cc2
Browse repository at this point
Refactor run_multiprocess
Siddharth Krishna (siddharth-krishna)
committed
7ab770d
View commit details
Copy full SHA for 7ab770d
Browse repository at this point
Make run_multiprocess take Functions not nn.Modules
Siddharth Krishna (siddharth-krishna)
committed
4e37ce3
View commit details
Copy full SHA for 4e37ce3
Browse repository at this point
End-of-file newlines
Siddharth Krishna (siddharth-krishna)
committed
9576811
View commit details
Copy full SHA for 9576811
Browse repository at this point
Commits on Apr 27, 2021
Black
Siddharth Krishna (siddharth-krishna)
committed
c4ca1a0
View commit details
Copy full SHA for c4ca1a0
Browse repository at this point
Parametrize pytest test_owt
Siddharth Krishna (siddharth-krishna)
committed
a095665
View commit details
Copy full SHA for a095665
Browse repository at this point
Revert "Fix bug: constructing variadic ops when specifying output_values"
Show description for 3279ca1
Siddharth Krishna (siddharth-krishna)
committed
3279ca1
View commit details
Copy full SHA for 3279ca1
Browse repository at this point
Fix Op constructor: better handling of variadic ops/pre-created outputs
Siddharth Krishna (siddharth-krishna)
committed
c84c61d
View commit details
Copy full SHA for c84c61d
Browse repository at this point
Commits on Apr 28, 2021
Add support for Relu
Siddharth Krishna (siddharth-krishna)
committed
bb604f6
View commit details
Copy full SHA for bb604f6
Browse repository at this point
Commits on Apr 29, 2021
Add DP test
Siddharth Krishna (siddharth-krishna)
committed
8d8bbda
View commit details
Copy full SHA for 8d8bbda
Browse repository at this point
Commits on Apr 30, 2021
Backend: run and time on GPU
Siddharth Krishna (siddharth-krishna)
committed
42751da
View commit details
Copy full SHA for 42751da
Browse repository at this point
Refactor grid_search for interactive use
Siddharth Krishna (siddharth-krishna)
committed
f73d33b
View commit details
Copy full SHA for f73d33b
Browse repository at this point
Commits on May 2, 2021
Timing code for CPUs
Siddharth Krishna (siddharth-krishna)
committed
20bae75
View commit details
Copy full SHA for 20bae75
Browse repository at this point
Handle ops with multiple outputs
Siddharth Krishna (siddharth-krishna)
committed
8e8513d
View commit details
Copy full SHA for 8e8513d
Browse repository at this point
Add support for all MLP training ops
Siddharth Krishna (siddharth-krishna)
committed
6f3ae59
View commit details
Copy full SHA for 6f3ae59
Browse repository at this point
Type inference: fix function name bug
Siddharth Krishna (siddharth-krishna)
committed
49d385c
View commit details
Copy full SHA for 49d385c
Browse repository at this point
Prettyprint: support for printing FunctionMakers
Siddharth Krishna (siddharth-krishna)
committed
2cce7c3
View commit details
Copy full SHA for 2cce7c3
Browse repository at this point
Fix backend op implementations
Siddharth Krishna (siddharth-krishna)
committed
91f5ecf
View commit details
Copy full SHA for 91f5ecf
Browse repository at this point
Convert per-rank fns to Modules inside each thread
Siddharth Krishna (siddharth-krishna)
committed
172a5db
View commit details
Copy full SHA for 172a5db
Browse repository at this point
Per-rank projector: remove types
Siddharth Krishna (siddharth-krishna)
committed
2ed3cf7
View commit details
Copy full SHA for 2ed3cf7
Browse repository at this point
Add some more tests
Siddharth Krishna (siddharth-krishna)
committed
7eee9f9
View commit details
Copy full SHA for 7eee9f9
Browse repository at this point
Default number of repetitions = 1
Siddharth Krishna (siddharth-krishna)
committed
78da37d
View commit details
Copy full SHA for 78da37d
Browse repository at this point
DHP transform: return separate init_fn and transformed fn
Siddharth Krishna (siddharth-krishna)
committed
56e05dc
View commit details
Copy full SHA for 56e05dc
Browse repository at this point
Run pytest with root dir included in PYTHONPATH
Siddharth Krishna (siddharth-krishna)
committed
78bc1e9
View commit details
Copy full SHA for 78bc1e9
Browse repository at this point
Grid search: remove unused imports
Siddharth Krishna (siddharth-krishna)
committed
8a288b6
View commit details
Copy full SHA for 8a288b6
Browse repository at this point
Revert unintended changes
Siddharth Krishna (siddharth-krishna)
committed
4b2b301
View commit details
Copy full SHA for 4b2b301
Browse repository at this point
Move run_pytorch to backend.torch
Siddharth Krishna (siddharth-krishna)
committed
75ec41a
View commit details
Copy full SHA for 75ec41a
Browse repository at this point
Commits on May 5, 2021
Interpret Function instead of creating fx.Graph
Siddharth Krishna (siddharth-krishna)
committed
6dbf57c
View commit details
Copy full SHA for 6dbf57c
Browse repository at this point
Prettyprint attributes as Python-style kwargs
Siddharth Krishna (siddharth-krishna)
committed
4a77383
View commit details
Copy full SHA for 4a77383
Browse repository at this point
Use broadcast with pairwise groups for send/recv on GPUs
Siddharth Krishna (siddharth-krishna)
committed
e8735c3
View commit details
Copy full SHA for e8735c3
Browse repository at this point
Move new tensors to GPU in each op, outputs back to CPU
Siddharth Krishna (siddharth-krishna)
committed
fd2e7b1
View commit details
Copy full SHA for fd2e7b1
Browse repository at this point
Commits on May 6, 2021
Add a mock multiprocess backend for debugging
Siddharth Krishna (siddharth-krishna)
committed
31031cc
View commit details
Copy full SHA for 31031cc
Browse repository at this point
Use spawn start method for multiprocessing
Siddharth Krishna (siddharth-krishna)
committed
5d99a62
View commit details
Copy full SHA for 5d99a62
Browse repository at this point
Fix MLP DHP tests
Siddharth Krishna (siddharth-krishna)
committed
2c8852a
View commit details
Copy full SHA for 2c8852a
Browse repository at this point
Revert "Fix MLP DHP tests"
Show description for 1d54fea
Siddharth Krishna (siddharth-krishna)
committed
1d54fea
View commit details
Copy full SHA for 1d54fea
Browse repository at this point
Fix MLP DHP tests for real
Siddharth Krishna (siddharth-krishna)
committed
18eaa08
View commit details
Copy full SHA for 18eaa08
Browse repository at this point
Commits on May 7, 2021
Add code to plot grid search results
Siddharth Krishna (siddharth-krishna)
committed
0ce2fdb
View commit details
Copy full SHA for 0ce2fdb
Browse repository at this point
Don't use globals while multiprocessing
Siddharth Krishna (siddharth-krishna)
committed
5588b87
View commit details
Copy full SHA for 5588b87
Browse repository at this point
Fix mock backend, use_gpu=False by default
Siddharth Krishna (siddharth-krishna)
committed
7d714d4
View commit details
Copy full SHA for 7d714d4
Browse repository at this point
Partial grid search on 4 devices
Siddharth Krishna (siddharth-krishna)
committed
8646842
View commit details
Copy full SHA for 8646842
Browse repository at this point
Commits on May 10, 2021
Support collectives between a subset of ranks
Siddharth Krishna (siddharth-krishna)
committed
1db0736
View commit details
Copy full SHA for 1db0736
Browse repository at this point
Commits on May 11, 2021
Bug fixes for distributed process groups (#24)
Show description for a79ab67
Keshav Santhanam (santhnm2)
authored
a79ab67
View commit details
Copy full SHA for a79ab67
Browse repository at this point
Commits on May 13, 2021
Debugging MLP deadlock
Siddharth Krishna (siddharth-krishna)
committed
0c0f7f5
View commit details
Copy full SHA for 0c0f7f5
Browse repository at this point
Enable grid search test again
Siddharth Krishna (siddharth-krishna)
committed
a5c8b34
View commit details
Copy full SHA for a5c8b34
Browse repository at this point
Commits on May 14, 2021
Fix collective projector
Keshav Santhanam (santhnm2)
committed
33cefce
View commit details
Copy full SHA for 33cefce
Browse repository at this point
Commits on May 20, 2021
Some comments, debugging code, cuda sync earlier
Siddharth Krishna (siddharth-krishna)
committed
ca68ec4
View commit details
Copy full SHA for ca68ec4
Browse repository at this point
Projector for gather
Siddharth Krishna (siddharth-krishna)
committed
b31209d
View commit details
Copy full SHA for b31209d
Browse repository at this point
Commits on May 25, 2021
Don't save input/outputs to file in torch backend
Siddharth Krishna (siddharth-krishna)
committed
54ce0d2
View commit details
Copy full SHA for 54ce0d2
Browse repository at this point
Remove unnecessary global
Siddharth Krishna (siddharth-krishna)
committed
5d1b63f
View commit details
Copy full SHA for 5d1b63f
Browse repository at this point
Free tensors after use
Siddharth Krishna (siddharth-krishna)
committed
3466627
View commit details
Copy full SHA for 3466627
Browse repository at this point
Commits on May 26, 2021
Map DistIR devices to pytorch backend ranks
Siddharth Krishna (siddharth-krishna)
committed
7d16c90
View commit details
Copy full SHA for 7d16c90
Browse repository at this point
Fix tests
Siddharth Krishna (siddharth-krishna)
committed
8c4f5e1
View commit details
Copy full SHA for 8c4f5e1
Browse repository at this point
Some documentation and cleanup
Siddharth Krishna (siddharth-krishna)
committed
79d4a65
View commit details
Copy full SHA for 79d4a65
Browse repository at this point
Fix comment
Siddharth Krishna (siddharth-krishna)
committed
da7ff5d
View commit details
Copy full SHA for da7ff5d
Browse repository at this point
Remove experiment code and dead code
Siddharth Krishna (siddharth-krishna)
committed
72589b2
View commit details
Copy full SHA for 72589b2
Browse repository at this point
Commits on Jun 1, 2021
Add ONNX exporter
Keshav Santhanam (santhnm2)
committed
849ab4e
View commit details
Copy full SHA for 849ab4e
Browse repository at this point
Remove singleton types
Keshav Santhanam (santhnm2)
committed
054ce0a
View commit details
Copy full SHA for 054ce0a
Browse repository at this point
Commits on Jun 2, 2021
Add test to verify that out-of-opset ops throw error
Keshav Santhanam (santhnm2)
committed
fab0020
View commit details
Copy full SHA for fab0020
Browse repository at this point
You can’t perform that action at this time.
