Comments (1)
Yep! Triton 1.0 has a bunch of API differences with 0.3. Most of them are summarized here: https://triton-lang.org/getting-started/tutorials/03-matrix-multiplication.html. Basically, three things to keep in mind:
- triton.kernel takes a device: torch.device argument in the constructor
- pointer arguments have to be passed explicitly using
torch.tensor.data_ptr()
. No more implicit conversion from torch.tensor to pointers. - auto-tuning is done slightly differently.
Hope it helps!
from triton.
Related Issues (20)
- How do I compile a Triton program? HOT 8
- Compile error with fp8 block pointer usage
- Jitting Error: can't pass bfloat16 as a tl.dype to my kernel! HOT 1
- Segmentation fault when DataLoader processes are launched after compiling Triton kernels HOT 6
- error when call tl.sort for data of torch.bfloat16 HOT 2
- Segfault on mixed mm HOT 4
- How to get the type of data from its ptr? HOT 1
- @LyricZhao Can you add a tutorial for the tile pointer? I think that could be helpful. HOT 1
- [Problem][Porposal] How does `make_block_ptr` work HOT 2
- FP8 conversion on SM89 (Ada Lovelace) fails with outdated PTX version HOT 4
- Initializing runtime driver overwrites root logger's verbosity level
- About Int8 Matrix Multiplication
- Why not allow JITFunction as parameter to another JITFunction(high-order jit function)?
- Add support for triangular solve operation.
- Inconsistency between constants as arguments and captured globals HOT 3
- The relationship between operator running time and number of runs HOT 2
- How to figure out layout/order for loads via explicit Tma? HOT 10
- Support for tl shift operator on tensor HOT 3
- tl.clamp bug : AttributeError("module 'triton.language' has no attribute 'clamp'") HOT 4
- fp8 tensor core support on h100 HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from triton.