Commit 9b0f4e0

zhangchenyi_dl

authored and

committed

[Metax] support rejection

1 parent e8b51ed commit 9b0f4e0Copy full SHA for 9b0f4e0

3 files changed

+523

-345

lines changed

custom_ops
- gpu_ops/sample_kernels
  - sampling.cuh
  - utils.cuh
- setup_ops.py

+523

-345

lines changed

Comments

(0)