MCPcopy Create free account
hub / github.com/pytorch/pytorch / support_8bit

Method support_8bit

caffe2/python/layers/sparse_lookup.py:269–274  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

267 return [self.w]
268
269 def support_8bit(self):
270 # Rowwise quantization makes sense only if shape it's 2D matrix with
271 # second dimension >= 8
272 if len(self.shape) != 2 or self.shape[1] < 8:
273 return False
274 return True
275
276 def get_8bits_compatible_parameters(self, fused=True):
277 if not self.support_8bit():

Callers 2

add_opsMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected