Skip to content

Instantly share code, notes, and snippets.

#!/usr/bin/env python3
"""Print pure-PyTorch MoE error caused by FP8-quantizing the middle activation."""
from __future__ import annotations
import argparse
from dataclasses import dataclass
import torch
from torch.nn import functional as F
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.