strisunshinewentingwang/pytorch

Pull Requests Commits

zasdfgbnm committed 3 years ago

18e9543c

zasdfgbnm committed 3 years ago

853f6899

Add note on ifdefing based on CUDA_VERSION for ROCm path (#62850)

jithunnair-amd committed 3 years ago

730ce29b

Small fixes to the Contributing.txt (#63385)

Gamrix committed 3 years ago

b5b9ce14

Back out "Temporary fix for remote gpu execution issue" (#63983)

Rong Rong (AI Infra) committed 3 years ago

52ebe7e1

Shape Propagation Pass: Fix AdaptiveAveragePooling2d (#63629)

priyaramani committed 3 years ago

5b548f6f

Move existing target determinator to tools (#63809)

driazati committed 3 years ago

ab5cf5a1

Add a comment on the potential implicit type up-casting (#63905)

Yi Wang committed 3 years ago

7edeead7

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

zasdfgbnm committed 3 years ago

505c715b

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

zasdfgbnm committed 3 years ago

3277d158

[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

zasdfgbnm committed 3 years ago

aa89b4c8

zasdfgbnm committed 3 years ago

5f561f22

[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

zasdfgbnm committed 3 years ago

12334bce

zasdfgbnm committed 3 years ago

d19c85e8

zasdfgbnm committed 3 years ago

25896c3e

Minor OptionalTensorRef updates (#63611)

peterbell10 committed 3 years ago

c1dfd587

zasdfgbnm committed 3 years ago

fcbeb2f1

Newer Older