[OpOptimization]add TVM batch norm 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 8 months ago

zwshan commented 10 months ago

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