Skip to content

Commit 5610b1b

Browse files
Deprecate igemm, batched_igemm, check_matmul APIs
1 parent 7ec290a commit 5610b1b

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

bitsandbytes/functional.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
import numpy as np
1212
import torch
1313
from torch import Tensor
14+
from typing_extensions import deprecated
1415

1516
from bitsandbytes.utils import pack_dict_to_tensor, unpack_tensor_to_dict
1617

@@ -1211,6 +1212,7 @@ def optimizer_update_8bit_blockwise(
12111212
)
12121213

12131214

1215+
@deprecated("This function is deprecated and will be removed in a future release.", category=FutureWarning)
12141216
def check_matmul(A, B, out, transposed_A, transposed_B, expected_type=torch.int8):
12151217
if not torch.cuda.is_initialized():
12161218
torch.cuda.init()
@@ -1332,6 +1334,7 @@ def gemv_4bit(
13321334
)
13331335

13341336

1337+
@deprecated("This function is deprecated and will be removed in a future release.", category=FutureWarning)
13351338
def igemm(
13361339
A: Tensor,
13371340
B: Tensor,
@@ -1433,6 +1436,7 @@ def igemm(
14331436
return out
14341437

14351438

1439+
@deprecated("This function is deprecated and will be removed in a future release.", category=FutureWarning)
14361440
def batched_igemm(
14371441
A: Tensor,
14381442
B: Tensor,

0 commit comments

Comments
 (0)