[OpOptimization]add TVM batchnorm cpu gpu version

buddy-compiler / buddy-benchmark

Benchmark Framework for Buddy Projects

Apache License 2.0

44 stars 33 forks source link

Closed zwshan closed 10 months ago

zwshan commented 10 months ago

Implemented CPU and GPU versions for benchmarking optimized batch norm in TVM.