onnxruntime
Using cublasGemmBatchedEx/cublasGemmStridedBatchedEx for training
#4731

Merged

Using cublasGemmBatchedEx/cublasGemmStridedBatchedEx for training #4731

weixingzhang merged 2 commits into master from wezhan/matmul

use cublas extenstion API for fp16

dbcaf2e8

Using cublasGemmBatchedEx/cublasGemmStridedBatchedEx for training

da8b4800

weixingzhang requested a review 5 years ago

weixingzhang added training

weixingzhang requested a review from

wschin 5 years ago

weixingzhang requested a review from

SherlockNoMad 5 years ago

weixingzhang requested a review from

souptc 5 years ago

wschin commented on 2020-08-07

wschin approved these changes on 2020-08-07

edgchen1 commented on 2020-08-08

SherlockNoMad approved these changes on 2020-08-10

weixingzhang merged afa89566 into master 5 years ago

weixingzhang deleted the wezhan/matmul branch 5 years ago

Reviewers

SherlockNoMad

wschin

edgchen1

souptc

Assignees

No one assigned

Labels

training

Milestone

No milestone