KernelGenWgsl.h

浏览该文件的文档.

#pragma once
 
#include "common/Result.h"
#include "tensor/KernelGen.h"
 
namespace eve::tensor {
enum class KernelVariant { Default, TiledMatMul };
 
[[nodiscard]] Result<KernelSpec> generateWgslKernel(const Graph &graph, const FusedGroup &group,
                                                    KernelVariant variant = KernelVariant::Default);
}  // namespace eve::tensor