squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix by adding public (#1753)

Browse Source
This commit is contained in:
Xinyu Yang 2024-10-24 00:45:58 +08:00 committed by GitHub
parent ea69cc2849
commit b0c09ed077
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/cutlass/gemm/kernel/gemm_universal_with_visitor.h
View File

@ -52,12 +52,12 @@ template <
typename Epilogue, ///! Epilogue typename Epilogue, ///! Epilogue
typename ThreadblockSwizzle_ ///! Threadblock swizzling function typename ThreadblockSwizzle_ ///! Threadblock swizzling function
> >
class GemmWithEpilogueVisitor: GemmUniversal<Mma,Epilogue, ThreadblockSwizzle_> { class GemmWithEpilogueVisitor: public GemmUniversal<Mma, Epilogue, ThreadblockSwizzle_> {
public: public:
using ThreadblockSwizzle = ThreadblockSwizzle_; using ThreadblockSwizzle = ThreadblockSwizzle_;
using Base = GemmUniversal<Mma,Epilogue, ThreadblockSwizzle>; using Base = GemmUniversal<Mma, Epilogue, ThreadblockSwizzle>;
using Base::Base; using Base::Base;
using FusionCallbacks = typename Epilogue::FusionCallbacks; using FusionCallbacks = typename Epilogue::FusionCallbacks;