Jerry Zhang 938803df94 Add bfloat16 support for per tensor/channel cpu/cuda fake quantize ops (#139306) ... Summary: Fixes https://fb.workplace.com/groups/2240361332735959/permalink/8190736677698365 Test Plan: buck2 test 'fbcode//mode/dev' fbcode//caffe2/test/quantization:test_quantization -- --exact 'caffe2/test/quantization:test_quantization - test_forward_per_channel_cachemask_cpu (caffe2.test.quantization.core.test_workflow_ops.TestFakeQuantizeOps)' buck2 test 'fbcode//mode/dev-nosan' fbcode//caffe2/test/quantization:test_quantization -- --exact 'caffe2/test/quantization:test_quantization - test_forward_per_tensor_cachemask_cpu (caffe2.test.quantization.core.test_workflow_ops.TestFakeQuantizeOps)' buck2 test 'fbcode//mode/dev-nosan' fbcode//caffe2/test/quantization:test_quantization -- --exact 'caffe2/test/quantization:test_quantization - test_forward_per_channel_cachemask_cuda (caffe2.test.quantization.core.test_workflow_ops.TestFakeQuantizeOps)' buck2 test 'fbcode//mode/dev-nosan' fbcode//caffe2/test/quantization:test_quantization -- --exact 'caffe2/test/quantization:test_quantization - test_forward_per_channel_cachemask_cpu (caffe2.test.quantization.core.test_workflow_ops.TestFakeQuantizeOps)' Differential Revision: D65221710 Pull Request resolved: https://github.com/pytorch/pytorch/pull/139306 Approved by: https://github.com/navsud		2024-10-31 20:41:15 +00:00
..
experimental
__init__.py
test_backend_config.py
test_docs.py
test_quantized_functional.py
test_quantized_module.py
test_quantized_op.py	[AOTI] add C shim for QConvPointWise (#138540)	2024-10-31 02:03:01 +00:00
test_quantized_tensor.py
test_top_level_apis.py
test_utils.py	[reland] Add int1 to int7 dtypes (#137928)	2024-10-18 02:02:08 +00:00
test_workflow_module.py
test_workflow_ops.py	Add bfloat16 support for per tensor/channel cpu/cuda fake quantize ops (#139306)	2024-10-31 20:41:15 +00:00