frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ET-Vulkan] ExecuTorch Vulkan floor_div (#118428)

Browse Source 
Summary: Add a new operator "floor_div" to ET-Vulkan.

Test Plan:
```
[yipjustin@7777.od ~/fbcode (b32108c6c)]$ buck2 test fbcode//executorch/backends/vulkan/test:test_vulkan_delegate --
File changed: fbcode//executorch/backends/vulkan/test/test_vulkan_delegate.py
Buck UI: https://www.internalfb.com/buck2/90290e5b-d47e-4cac-bc63-9939cc210d1f
Test UI: https://www.internalfb.com/intern/testinfra/testrun/3940649890839142
Network: Up: 2.8KiB  Down: 0B  (reSessionID-e7425cc1-0987-46d8-a7bf-418a660bee5b)
Jobs completed: 19. Time elapsed: 42.6s.
Cache hits: 0%. Commands: 1 (cached: 0, remote: 0, local: 1)
Tests finished: Pass 7. Fail 0. Fatal 0. Skip 0. Build failure 0
```

Reviewed By: SS-JIA

Differential Revision: D53072722

Pull Request resolved: https://github.com/pytorch/pytorch/pull/118428
Approved by: https://github.com/SS-JIA
This commit is contained in:
Justin Yip
2024-01-27 11:20:52 +00:00
committed by PyTorch MergeBot
parent 8d790abab9
commit 26473460a4
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aten/src/ATen/native/vulkan/impl/Arithmetic.cpp
View File

@ -16,6 +16,8 @@ api::ShaderInfo get_shader(const OpType type) {
return VK_KERNEL(mul);
case OpType::DIV:
return VK_KERNEL(div);
case OpType::FLOOR_DIV:
return VK_KERNEL(floor_divide);
}
VK_THROW("Invalid OpType");
}

1
aten/src/ATen/native/vulkan/impl/Arithmetic.h
View File

@ -14,6 +14,7 @@ enum class OpType : uint32_t {
SUB,
MUL,
DIV,
FLOOR_DIV,
};
api::ShaderInfo get_shader(const OpType type);