flashinfer.cake_rmsnorm_train.is_cake_rmsnorm_train_supported

flashinfer.cake_rmsnorm_train.is_cake_rmsnorm_train_supported(device: device, hidden: int | None = None, *, deterministic: bool = True) → bool

Whether device (and optionally hidden) has an exported route.

deterministic=True answers for the deterministic package (flashinfer.cake_rmsnorm_det_train.is_cake_rmsnorm_det_train_supported()), False for this package’s non-deterministic routes. The answer does not depend on torch.use_deterministic_algorithms.