Skip to content

Commit fda86cf

Browse files
authored
Add aten::mvlgamma (#1041)
CUDA dispatches to CPU as well. - `mvlgamma.out` - `mvlgamma` - `mvlgamma_`
1 parent 3e475c0 commit fda86cf

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

yaml/native/native_functions.yaml

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -185,6 +185,25 @@
185185
tags: pointwise
186186
# multiply, alias for mul
187187

188+
- func: mvlgamma.out(Tensor self, int p, *, Tensor(a!) out) -> Tensor(a!)
189+
dispatch:
190+
XPU: mvlgamma_out
191+
tags: pointwise
192+
193+
- func: mvlgamma(Tensor self, int p) -> Tensor
194+
device_check: NoCheck # TensorIterator
195+
variants: function, method
196+
dispatch:
197+
CompositeExplicitAutograd: mvlgamma
198+
tags: pointwise
199+
200+
- func: mvlgamma_(Tensor(a!) self, int p) -> Tensor(a!)
201+
device_check: NoCheck # TensorIterator
202+
variants: method
203+
dispatch:
204+
CompositeExplicitAutograd: mvlgamma_
205+
tags: pointwise
206+
188207
- func: div.Tensor(Tensor self, Tensor other) -> Tensor
189208
device_check: NoCheck # TensorIterator
190209
variants: function, method

0 commit comments

Comments
 (0)