summaryrefslogtreecommitdiff
path: root/bitsandbytes/autograd/_functions.py
diff options
context:
space:
mode:
authorTim Dettmers <tim.dettmers@gmail.com>2022-08-08 09:13:22 -0700
committerTim Dettmers <tim.dettmers@gmail.com>2022-08-08 09:13:22 -0700
commitf9cbe2fe99c805dcca934c66677951f428d3b3e2 (patch)
treefaa06c4f066517312221e3fad8694fdae890daba /bitsandbytes/autograd/_functions.py
parent62441815bc733c9e75d32dd65305a16aaebd317a (diff)
Fixed prod Python < 3.7 compatibility in function.py.
Diffstat (limited to 'bitsandbytes/autograd/_functions.py')
-rw-r--r--bitsandbytes/autograd/_functions.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/bitsandbytes/autograd/_functions.py b/bitsandbytes/autograd/_functions.py
index a5446b7..01e7073 100644
--- a/bitsandbytes/autograd/_functions.py
+++ b/bitsandbytes/autograd/_functions.py
@@ -6,6 +6,7 @@ import bitsandbytes.functional as F
from dataclasses import dataclass
from functools import reduce # Required in Python 3
+# math.prod not compatible with python < 3.8
def prod(iterable):
return reduce(operator.mul, iterable, 1)