From ed774769b4787f22a7ee4df46d985925bbaca77d Mon Sep 17 00:00:00 2001 From: Prasanth Nunna Date: Thu, 9 May 2024 21:59:14 +0000 Subject: [PATCH] Sync with upstream --- tests/test_functional.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/tests/test_functional.py b/tests/test_functional.py index 99cbf6f75..21dae281a 100644 --- a/tests/test_functional.py +++ b/tests/test_functional.py @@ -11,8 +11,15 @@ import bitsandbytes as bnb from bitsandbytes import functional as F +from tests.helpers import ( + BOOLEAN_TUPLES, + TRUE_FALSE, + describe_dtype, + get_blocksizes, + get_test_dims, + id_formatter, +) from bitsandbytes.cextension import BNB_HIP_VERSION, HIP_ENVIRONMENT, ROCM_GPU_ARCH -from tests.helpers import BOOLEAN_TUPLES, TRUE_FALSE, describe_dtype, get_blocksizes, get_test_dims, id_formatter torch.set_printoptions(precision=5, sci_mode=False, linewidth=120, edgeitems=20, threshold=10000) k = 20