forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
cudnn frontend v0.6.0 had an issue. we upgrade to cudnn frontend to v0.6.1. see reverted PR: pytorch#75481 Pull Request resolved: pytorch#75615 Approved by: https://github.com/jerryzh168
- Loading branch information
1 parent
143f7cc
commit 25aa251
Showing
1 changed file
with
1 addition
and
1 deletion.
There are no files selected for viewing
Submodule cudnn_frontend
updated
21 files
+10 −0 | CMakeLists.txt | |
+1 −1 | Doxyfile | |
+8 −1 | README.md | |
+29 −33 | include/cudnn_frontend_Engine.h | |
+3 −1 | include/cudnn_frontend_EngineConfigGenerator.h | |
+11 −0 | include/cudnn_frontend_EngineFallbackList.h | |
+172 −0 | include/cudnn_frontend_ExecutionPlan.h | |
+1 −1 | include/cudnn_frontend_ExecutionPlanCache.h | |
+284 −3 | include/cudnn_frontend_Heuristics.h | |
+484 −19 | include/cudnn_frontend_Operation.h | |
+38 −2 | include/cudnn_frontend_PointWiseDesc.h | |
+2 −2 | include/cudnn_frontend_Reorder_Tensor.h | |
+18 −11 | include/cudnn_frontend_VariantPack.h | |
+1 −0 | include/cudnn_frontend_get_plan.h | |
+24 −44 | samples/conv_sample.cpp | |
+1,072 −37 | samples/fusion_sample.cpp | |
+78 −0 | samples/fusion_sample.h | |
+33 −0 | samples/helpers.cpp | |
+3 −0 | samples/helpers.h | |
+65 −35 | samples/mha_sample.cpp | |
+251 −8 | samples/test_list.cpp |