OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
release/1.6
pytorch/test/quantization
History
Nikita Shulga 7fa9b2923b
quantizer.cpp: fix cuda memory pinning (#41139) (#41194) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41139

Fixes the test case in https://github.com/pytorch/pytorch/issues/41115
by using PyTorch's CUDA allocator instead of the old Caffe2 one.

Test Plan:
run the test case from the issue:
https://gist.github.com/vkuzo/6d013aa1645cb986d0d4464a931c779b

let's run CI and see what it uncovers

Imported from OSS

Reviewed By: malfet

Differential Revision: D22438787

fbshipit-source-id: 0853b0115d198a99c43e6176aef34ea951bf5c2e

Co-authored-by: Vasiliy Kuznetsov <vasiliy@fb.com>
2020-07-09 14:06:11 -07:00
..
serialized [quant] Add save/load state_dict to quantized dynamic RNNs (#39105) 2020-05-28 10:37:38 -07:00
__init__.py Stopgap fix to determine_target predicate (#37934) 2020-05-06 14:05:14 -07:00
test_backward_compatibility.py [quant] Add save/load state_dict to quantized dynamic RNNs (#39105) 2020-05-28 10:37:38 -07:00
test_fusion_passes.py Add JIT fusion pass to fuse quantized add and relu. (#38897) 2020-05-27 14:16:57 -07:00
test_numeric_suite.py [PyTorch Numeric Suite] Add support for dynamic LSTM (#40065) 2020-06-20 07:00:13 -07:00
test_qat_module.py Updates assertEqual to require atol and rtol, removes positional atol (#38872) 2020-05-27 06:31:07 -07:00
test_quantize_jit.py [quant][graphmode][fix] Fold conv bn (#40865) (#40970) 2020-07-07 13:25:39 -07:00
test_quantize.py qat eager: remove unneeded modules (#40396) 2020-06-22 17:45:51 -07:00
test_quantized_functional.py Enabled per channel quantized static linear/conv (#37622) 2020-05-20 14:01:37 -07:00
test_quantized_module.py Add state dict for LSTM and RNNCell and helper functions for accessing weights and bias (#40333) 2020-06-22 17:41:07 -07:00
test_quantized_op.py Re-apply PyTorch pthreadpool changes (#40951) 2020-07-06 06:58:25 -07:00
test_quantized_tensor.py quantizer.cpp: fix cuda memory pinning (#41139) (#41194) 2020-07-09 14:06:11 -07:00
test_workflow_module.py [quant][bug] Histogram observer bug fix with min == max (#40310) 2020-06-22 10:05:10 -07:00