Skip to content
Permalink
Browse files

Enable 4bit RWQ SLS (#4009)

Summary:
Pull Request resolved: #4009

Not sure why these weren't enabled for the 0.4.2.5 release.

Reviewed By: yinghai

Differential Revision: D19397626

fbshipit-source-id: b1ff72e07a5f0e8e997021bcc70653a2e674f4d5
  • Loading branch information
jfix71 authored and facebook-github-bot committed Jan 15, 2020
1 parent 7d364cd commit bae79d6e59a7bddd41b37c535137abfe189e8a80
Showing with 2 additions and 4 deletions.
  1. +2 −1 lib/Backends/NNPI/NNPI.cpp
  2. +0 −3 lib/Backends/NNPI/tests/NNPIOperatorTest.cpp
@@ -243,7 +243,8 @@ bool NNPIBackend::isOpSupported(const NodeInfo &NI) const {
auto resultK =
NI.getOutElemTy(FusedRowwiseQuantizedSparseLengthsSumNode::ResultIdx);
return (dataK == ElemKind::UInt8FusedQTy ||
dataK == ElemKind::UInt8FusedFP16QTy) &&
dataK == ElemKind::UInt8FusedFP16QTy ||
dataK == ElemKind::UInt4FusedFP16QTy) &&
(resultK == ElemKind::FloatTy || resultK == ElemKind::Float16Ty) &&
(indicesK == ElemKind::Int64ITy) && (lengthsK == ElemKind::Int32ITy);
}
@@ -157,9 +157,6 @@ struct EmulatorOnlyTests {
"Exp/0",
"FloatArgMaxKeepDim/0",
"FloatArgMaxNoKeepDim/0",
"FusedRowwiseQuantizedSLWSTwoColumn_Fused4Bit_Float16_AccumFloat16/0",
"FusedRowwiseQuantizedSparseLengthsSum_Fused4Bit_Float16_"
"AccumFloat16/0",
"FusedRowwiseQuantizedSparseLengthsWeightedSum_ConvertedFloat16_"
"back_",
"FusedRowwiseQuantizedSparseLengthsWeightedSum_ConvertedFloat16_"

0 comments on commit bae79d6

Please sign in to comment.
You can’t perform that action at this time.