[ExecuTorch][WebGPU] Add floor_divide op (aten.div.Tensor_mode)#21207
[ExecuTorch][WebGPU] Add floor_divide op (aten.div.Tensor_mode)#21207JCNTH wants to merge 1 commit into
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/21207
Note: Links to docs will display an error until the docs builds have been completed. ❗ 1 Active SEVsThere are 1 currently active SEVs. If your PR is affected, please view them below: ❌ 47 New Failures, 3 Unrelated FailuresAs of commit 13cd5cd with merge base 266e0dc ( NEW FAILURES - The following jobs have failed:
FLAKY - The following jobs failed but were likely due to flakiness present on trunk:
BROKEN TRUNK - The following job failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
This PR needs a
|
psiddh
left a comment
There was a problem hiding this comment.
Approving full WebGPU stack
Stack from ghstack (oldest at bottom):
Adds
div.Tensor_mode(floor rounding) asfloor(in1 / in2)with NumPy broadcasting, mirroringmul+ Vulkan. Onlyrounding_mode='floor'is supported (fail-loud otherwise).Key changes:
runtime/ops/floor_divide/{BinaryOp.cpp, binary_floor_divide.wgsl}(+ generated_wgsl.h) —floor_divide_impl(args[in1, in2, rounding_mode, out]); broadcastTensorMetapath + dual-operand resize hook (mirrorsmul); rounding_mode + int guards.CMakeLists.txt—runtime/ops/floor_divide/BinaryOp.cppinWEBGPU_SRCS.Differential Revision: D112257613