Allow no-bias MKLDNN Linear call (#43703)

bwasti · facebook-github-bot · commit 14ebb2c67cd1 · 2020-09-02T08:54:50.000-07:00
Summary: MKLDNN linear incorrectly assumes that bias is defined and will fail for no-bias calls. Pull Request resolved: #43703 Reviewed By: glaringlee Differential Revision: D23373182 Pulled By: bwasti fbshipit-source-id: 1e817674838a07d237c02eebe235c386cf5b191e
diff --git a/aten/src/ATen/native/mkldnn/Linear.cpp b/aten/src/ATen/native/mkldnn/Linear.cpp
@@ -32,7 +32,8 @@ Tensor mkldnn_linear(
       "mkldnn_linear: input needs to has dim at least 2, input dim ", self.dim());
   TORCH_CHECK(self.is_mkldnn(),
       "mkldnn_linear: input needs to be mkldnn layout");
-  TORCH_CHECK(weight.is_mkldnn() && bias.is_mkldnn(),
+  TORCH_CHECK(
+      weight.is_mkldnn() && (!bias.defined() || bias.is_mkldnn()),
       "mkldnn_linear: weight and bias need to be mkldnn layout");
 
   // reshape first if input dim is greater than 2 and the reshape will cost a memory copy.
