Jason Ansel
Jason Ansel
This decomposition: ``` @register_decomposition([aten.transpose], decompositions) def transpose(x, dim0: int, dim1: int): ndim = x.ndimension() dims = list(range(ndim)) dims[dim0], dims[dim1] = dims[dim1], dims[dim0] return torch.permute(x, dims) ``` Gives this error: ```...
I've noticed that in the past few months the user experience for using TorchBench for correctness testing [TorchDynamo](https://github.com/facebookresearch/torchdynamo) has gotten much worse. TorchBench used to run quickly, and I could...
If you try to commit an empty file, you get the following error: ``` Traceback (most recent call last): File "/home/jansel/testrepo/venv/bin/git-bigstore", line 87, in args = parser.parse_args() File "/usr/lib/python2.7/argparse.py", line...
I have not tested the other backends. The first issue is the use of `tempfile.TemporaryFile` instead of `tempfile.NamedTemporaryFile`. The temporary file does not have a name on disk on most...
``` @triton.jit def kernel0(out_ptr2, xnumel, XBLOCK : tl.constexpr): xoffset = tl.program_id(0) * XBLOCK xindex = xoffset + tl.reshape(tl.arange(0, XBLOCK), [XBLOCK, 1]) xmask = xindex < xnumel tmp6 = 1.0 #...
```py import torch import triton import triton.language as tl from torch import empty_strided from triton import cdiv @triton.autotune( [ triton.Config( { "XBLOCK": 32, "YBLOCK": 32, # This segfaults: "ZBLOCK": 1...
Repro: ``` import torch import triton import triton.language as tl @triton.jit def kernel2(in_ptr0, out_ptr5, xnumel: tl.constexpr, XBLOCK: tl.constexpr): xoffset = tl.program_id(0) * XBLOCK xindex = xoffset + tl.reshape(tl.arange(0, XBLOCK), [XBLOCK])...
Repro: ``` from torch import empty_strided from triton import cdiv import torch import triton import triton.language as tl @triton.jit def kernel0(in_ptr0, in_ptr1, in_ptr2, in_ptr3, in_ptr4, out_ptr0, xnumel: tl.constexpr, ynumel: tl.constexpr,...
Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #90739 Differential Revision: [D41845303](https://our.internmc.facebook.com/intern/diff/D41845303/) **NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D41845303/)! cc @mlazos...
Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #92188 * __->__ #92187 cc @mlazos @soumith @voznesenskym @yanboliang @penguinwu @anijain2305 @EikanWang @jgong5 @Guobing-Chen @chunyuan-w @XiaobingSuper @zhuhaozhe @blzheng @Xia-Weiwen @wenzhe-nrv @jiayisunx @peterbell10 @desertfire