Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 3 additions & 5 deletions fbgemm_gpu/fbgemm_gpu/quantize_comm.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,13 +13,12 @@


import logging
from dataclasses import dataclass
from typing import TypeVar

_T = TypeVar("_T")

import torch

# fmt:skip
import fbgemm_gpu.quantize.quantize_ops # noqa: F401
from fbgemm_gpu.quantize_utils import (
bf16_to_fp32,
fp16_to_fp32,
Expand All @@ -33,9 +32,8 @@
)
from fbgemm_gpu.split_embedding_configs import SparseType
from torch.autograd.profiler import record_function # usort:skip
from dataclasses import dataclass

import fbgemm_gpu.quantize.quantize_ops # noqa F401
_T = TypeVar("_T")

logger: logging.Logger = logging.getLogger()

Expand Down
Loading