github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[BACKEND] Enable reduce with 3D tensors and added tests (#2460)

Browse Source
This commit is contained in:
Zahi Moudallal
2023-10-06 15:08:22 -07:00
committed by GitHub
parent a42d517021
commit be19cf3103
4 changed files with 60 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/triton/Analysis/Utility.h
View File

@@ -54,6 +54,8 @@ public:
SmallVector<unsigned> getScratchConfig();
SmallVector<unsigned> getOrderWithAxisAtBeginning();
unsigned getScratchSizeInBytes();
bool isSupportedLayout();