strisunshinewentingwang/pytorch

Pull Requests Commits

add BFloat16 support for bernoulli and Dropout on CPU (#56372)

mingfeima committed 3 years ago

b0782f0f

Update torch.distributed.run OMP_NUM_THREADS message to log.warning (#63953)

H-Huang committed 3 years ago

72995657

Fix ciflow/all label generation (#63954)

zhouzhuojie committed 3 years ago

3d4aabfc

Reformat run_test.py (#63808)

driazati committed 3 years ago

67d8e7b6

[Static Runtime] Added caching for the NNC code generated for Logit. (#63840)

navahgar committed 3 years ago

64d605ba

[Static Runtime] Added a variable for clamp in the NNC code for Logit. (#63839)

navahgar committed 3 years ago

dde07cad

[Static Runtime] Moved NNC operator definitions to separate files. (#63838)

navahgar committed 3 years ago

a2399a76

[Reland] Replacing the p.data acccess in utils with tensor.set_ . Passes both test_post_localSGD_optimizer_pari and test_periodic_model_averager tests (#63895)

Aayush Prakash committed 3 years ago

8a22d4fa

clean up engine.cpp thread state (#63115)

albanD committed 3 years ago

ab954cb0

[fx2trt] Check input device in TRTModule (#63893)

842974287 committed 3 years ago

c06dfd7c

bf16 Error message cleanup as well as addition of is_bf16_supported (#63798)

riship committed 3 years ago

6324d98e

[pruner] add getter for pruned outputs in base pruner (#63520)

kazhou committed 3 years ago

eebac462

[pruner] add support for pruning BatchNorm2d (#63519)

kazhou committed 3 years ago

83b132b1

Minor OptionalTensorRef updates (#63611)

peterbell10 committed 3 years ago

c1dfd587

Update CMake minimum version to 3.10 (#63660)

malfet committed 3 years ago

5ab356ff

Temporary fix for remote gpu execution issue (#63899)

Rong Rong (AI Infra) committed 3 years ago

34ed16ff

Fix bug in `check_empty_containers` (#63492)

Ansley Ussery committed 3 years ago

01c35115

Swap CUDA 11.1 and 11.3 in CI to make 11.1 periodic (#63900)

janeyx99 committed 3 years ago

8c897d25

[skip ci] Add generated comment to ruleset json (#63896)

zhouzhuojie committed 3 years ago

3926fdba

Revert D30526034: [pytorch][PR] compute reduction intermediate buffer size in elements

albanD committed 3 years ago

87a661c7

Revert D30384746: [fx2trt] Add a test for quantized resnet18

linbinyu committed 3 years ago

839eaa2e

[fx2trt] Add a test for quantized resnet18 (#63446)

jerryzh168 committed 3 years ago

10dfa58e

[quant][graphmode][fx] Make maxpool and flatten produce the reference pattern (#63501)

jerryzh168 committed 3 years ago

0301c3bc

[TensorExpr] LLVMCodegen: Use addFnAttr instead of addAttribute which was deleted. (#63886)

Mikhail Zolotukhin committed 3 years ago

d388a1a5

[qunat][graphmode][fx] Add a separate lower_to_native_backend function for relu (#62861)

jerryzh168 committed 3 years ago

c8527bc3

compute reduction intermediate buffer size in elements (#63885)

ngimel committed 3 years ago

e69a1398

TST Adds more modules into common module tests (#62999)

thomasjpfan committed 3 years ago

ba126df6

Allow arbitrary objects in state_dicts (#62976)

jbschlosser committed 3 years ago

544af391

TST Adds pickle testing for ModuleInfo (#63736)

thomasjpfan committed 3 years ago

58ef99bd

Re-apply: [nnc] Support thread level parallelism in fused kernels (#63776)

bertmaher committed 3 years ago

8dda299d

Newer Older