[Kernel] DBRX Triton MoE kernel H100

[ywang96]

Results from running benchmarks/benchmark_latency.py --model=databricks/dbrx-instruct --num-iters 50 --batch-size --trust-remote-code --tp <tp> --batch-size <bs>

Main branch

TP4
Batch size 8 (default): Avg latency: 3.383673385512084 seconds
Batch size 16: Avg latency: 3.910283356476575 seconds
Batch size 32: Avg latency: 5.204639058653266 seconds
Batch size 64: Avg latency: 5.740127996951341 seconds

TP8
Batch size 8: Avg latency: 2.3971098379790785 seconds
Batch size 16: Avg latency: 2.7930808463692665 seconds
Batch size 32: Avg latency: 3.329706186875701 seconds
Batch size 64: Avg latency: 3.84231577526778 seconds

This PR

TP4
Batch size 8: Avg latency: 3.2545747229084374 seconds
Batch size 16: Avg latency: 3.6656158388964832 seconds
Batch size 32: Avg latency: 3.978769546896219 seconds
Batch size 64: Avg latency: 4.446144629847258 seconds

TP8
Batch size 8: Avg latency: 2.305059498753399seconds
Batch size 16: Avg latency: 2.564341889191419 seconds 
Batch size 32: Avg latency: 2.7666698359698056 seconds
Batch size 64: Avg latency: 3.3472835789062083 seconds

cc @WoosukKwon @youkaichao

[WoosukKwon]

LGTM! Thanks!