intel / torch-xpu-ops

Apache License 2.0
30 stars 21 forks source link

Add aten::glu_jvp and its variants #1079

Closed gaopengff closed 1 week ago

gaopengff commented 1 week ago

Register glu_jvp and glu_jvp_backward for XPU. Only glu_jvp needs xpu kernel. glu_jvp_backward will use pytorch's common implementation just like cuda at code.

xytintel commented 1 week ago

image image