Skip to content

Commit

Permalink
Update on "[vulkan] Enable prepacked addmm for linear layers"
Browse files Browse the repository at this point in the history
Differential Revision: [D24908605](https://our.internmc.facebook.com/intern/diff/D24908605)

[ghstack-poisoned]
  • Loading branch information
SS-JIA committed Nov 12, 2020
1 parent 0ddb9f3 commit 6d9cc38
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/csrc/jit/passes/vulkan_rewrite.cpp
Expand Up @@ -50,7 +50,7 @@ void insertPrePackedLinearOp(std::shared_ptr<Graph>& graph) {
return (%res))";

const auto filter = [](const Match& match,
const std::unordered_map<std::string, Value*>& vmap) {
const std::unordered_map<std::string, Value*>& 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);
Expand Down

0 comments on commit 6d9cc38

Please sign in to comment.