From 6d9cc38b6dce1f248d8ec123a8f687e3779b9cbd Mon Sep 17 00:00:00 2001 From: ssjia Date: Thu, 12 Nov 2020 08:09:13 -0500 Subject: [PATCH] Update on "[vulkan] Enable prepacked addmm for linear layers" Differential Revision: [D24908605](https://our.internmc.facebook.com/intern/diff/D24908605) [ghstack-poisoned] --- torch/csrc/jit/passes/vulkan_rewrite.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch/csrc/jit/passes/vulkan_rewrite.cpp b/torch/csrc/jit/passes/vulkan_rewrite.cpp index f8f8794eccc1..4e381c47dae0 100644 --- a/torch/csrc/jit/passes/vulkan_rewrite.cpp +++ b/torch/csrc/jit/passes/vulkan_rewrite.cpp @@ -50,7 +50,7 @@ void insertPrePackedLinearOp(std::shared_ptr& graph) { return (%res))"; const auto filter = [](const Match& match, - const std::unordered_map& vmap) { + const std::unordered_map& vmap) { const auto& match_vmap = match.values_map; const auto linear_value = match_vmap.at(vmap.at("linear")); const auto func_name = graph_rewrite_helper::getFuncName(linear_value);