Skip to content

Commit 99a11ef

Browse files
Revert "Lift jagged -> padded dense forward / backward kernels from fbgemm_gpu (pytorch#125946)"
This reverts commit e2f0818. Reverted pytorch#125946 on behalf of https://github.com/clee2000 due to I think dr ci is wrong and the windows build failure is real https://hud.pytorch.org/pytorch/pytorch/commit/e2f081837f4276c1a6a37739bd28157f62004a06 https://github.com/pytorch/pytorch/actions/runs/9216826622/job/25357819877 ([comment](pytorch#125946 (comment)))
1 parent cfb374d commit 99a11ef

File tree

4 files changed

+0
-1108
lines changed

4 files changed

+0
-1108
lines changed

aten/src/ATen/native/native_functions.yaml

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -14636,16 +14636,6 @@
1463614636
NestedTensorCUDA: NestedTensor_to_padded_tensor_cuda
1463714637
autogen: to_padded_tensor.out
1463814638

14639-
- func: _jagged_to_padded_dense_forward(Tensor values, Tensor[] offsets, SymInt[] max_lengths, float padding_value=0.0) -> Tensor
14640-
variants: function
14641-
dispatch:
14642-
CUDA: _fbgemm_jagged_to_padded_dense_forward
14643-
14644-
- func: _padded_dense_to_jagged_forward(Tensor dense, Tensor[] offsets, SymInt? total_L=None) -> Tensor
14645-
variants: function
14646-
dispatch:
14647-
CUDA: _fbgemm_dense_to_jagged_forward_symint
14648-
1464914639
- func: _nested_tensor_softmax_with_shape(Tensor self, Tensor query) -> Tensor
1465014640
dispatch:
1465114641
NestedTensorCPU: NestedTensor_softmax_dropout

0 commit comments

Comments
 (0)