strisunshinewentingwang/pytorch

Pull Requests Commits

CUDA BFloat16 batched gemm (#45167)

zasdfgbnm committed 3 years ago

144dacd8

[JIT] Do not allow creating generics with None types (#44958)

malfet committed 3 years ago

989d877c

[reland][pytorch] refine dispatch keys in native_functions.yaml (1/N) (#45137)

ljk53 committed 3 years ago

0a9ac98b

[packaging] rstrip fix (#45166)

zdevito committed 3 years ago

25ed739a

torch.package - a way to package models and code (#45015)

zdevito committed 3 years ago

cb75adde

[RPC profiling] Don't wrap toHere() calls with profiling (#44655)

rohan-varma committed 3 years ago

d4a634c2

[RPC profiling] Allow disableProfiler() to be called from another thread. (#44653)

rohan-varma committed 3 years ago

70d2e4d1

Remove thread_local RecordFunctionGuard from profiler. (#44646)

rohan-varma committed 3 years ago

1bd6533d

CUDA BFloat16 pooling (#45151)

zasdfgbnm committed 3 years ago

67a19fec

[jit] gtestify test_argument_spec.cpp (#45019)

suo committed 3 years ago

666223df

[quant][graphmode][jit][api] Expose preserved_attrs from finalize to convert_jit (#44490)

jerryzh168 committed 3 years ago

f575df20

[JIT] Add default arguments for class types (#45098)

Meghan Lele committed 3 years ago

e0451199

[tensorexpr] Add flag to fuse with unknown shapes (#44401)

bwasti committed 3 years ago

ebde5a80

[caffe2] Support data types in shape hints (#45110)

Hao Lu committed 3 years ago

c0267c68

install ATen/native/cuda and hip headers (#45097)

jeffdaily committed 3 years ago

b98ac208

Relax CUDA architecture check (#45130)

malfet committed 3 years ago

2a37f3fd

[quant][graphmode][fx] Custom module support (#44766)

jerryzh168 committed 3 years ago

ccfbfe5e

[resubmit][FX] s/get_param/get_attr/ (#45147)

James Reed committed 3 years ago

7f4a27be

[PyTorch] Enable type check for autocast_test_lists (#45107)

hangjunxu committed 3 years ago

35cdb013

[JIT] Fix WithTest.test_with_exceptions (#45106)

Meghan Lele committed 3 years ago

cddcfde8

Clarify that 5-D 'bilinear' grid_sample is actually trilinear (#45090)

kurtamohler committed 3 years ago

d1c68a70

[FX] Make Graphs immutable and make GraphModule recompile after assigning graph (#44830)

James Reed committed 3 years ago

79fe794f

.circleci: Upgrade all xcode 9 workers to xcode 11 (#45153)

seemethere committed 3 years ago

def433bb

Fix type hint warnings for common_methods_invocations.py (#44971)

Viswesh Sankaran committed 3 years ago

a4ce3f41

Fix incorrect EnumValue serialization issue (#44891)

gmagogsfm committed 3 years ago

c253b101

[quant] Fix ConvTranspose mapping (#44844)

z-a-f committed 3 years ago

2b1f2588

[caffe2] Replace embedding conversion ops with fbgemm functions (#44843)

dskhudia committed 3 years ago

09aee06e

Support BFloat16 for binary logical operators on CUDA (#42485)

xuhdev committed 3 years ago

e2b40ce7

[pytorch] Add triplet margin loss with custom distance (#43680)

Mike Ruberry committed 3 years ago

ef885c10

Align casing in test_dispatch with dispatch keys. (#44933)

Ailing Zhang committed 3 years ago

10f28753

Newer Older