Enable bfloat16 for sparse_w16a16 (#18)

Author: mgoin

vllm/model_executor/layers/sparsity/sparse_w16a16.py

@@ -31,7 +31,7 @@ def get_name(cls) -> str:
 
     @classmethod
     def get_supported_act_dtypes(cls) -> List[torch.dtype]:
-        return [torch.half]
+        return [torch.float16, torch.bfloat16]
 
     @classmethod
     def get_min_capability(cls) -> int: