PR #1456 Switch to use or not einsum op. Needed for ORT

Switch to use or not einsum op. Needed for ORT #1456

jeffra merged 12 commits into deepspeedai:master from ganik/ort

Replace einsum op

ba8de46e

ganik requested a review from

awan-10 4 years ago

ganik requested a review from

cli99 4 years ago

ganik requested a review from

conglongli 4 years ago

ganik requested a review from

eltonzheng 4 years ago

ganik requested a review from

jeffra 4 years ago

ganik requested a review from

minjiaz 4 years ago

ganik requested a review from

niumanar 4 years ago

ganik requested a review from

RezaYazdaniAminabadi 4 years ago

ganik requested a review from

samyam 4 years ago

ganik requested a review from

ShadenSmith 4 years ago

ganik requested a review from

tjruwase 4 years ago

Re-write torch.einsum("sec,sm->ecm",..) with more performant way.

c907b924

Fix formatting

9704dfd2

Debug print statement to figure out shapes

4bc5dc4f

Log shapes for further debugging

179c3454

Fix tmp log file path.

aa3d1ee4

fix bbreak

a122e37d

more performant einsum rewrites

dc9f9604

fix format

dcd9c64f

Merge branch 'master' into ganik/ort

07f28d15

Set USE_EINSUM default to true.

0eb761c3

ganik changed the title ~~[WIP] Replace einsum op~~ Replace einsum op 4 years ago

ganik changed the title ~~Replace einsum op~~ Switch to use or not einsum op. Needed for ORT 4 years ago

Merge branch 'master' into ganik/ort

1d612b75

jeffra approved these changes on 2021-10-21

jeffra enabled auto-merge (squash) 4 years ago

jeffra merged 20bf1cc1 into master 4 years ago

Reviewers

jeffra

awan-10

cli99

conglongli

eltonzheng

minjiaz

niumanar

RezaYazdaniAminabadi

samyam

ShadenSmith

tjruwase

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

DeepSpeed Switch to use or not einsum op. Needed for ORT #1456 Merged

Switch to use or not einsum op. Needed for ORT #1456

DeepSpeed
Switch to use or not einsum op. Needed for ORT
#1456

Merged