Skip to content

Commit 672a111

Browse files
Hardcode84Diptorup Deb
authored andcommitted
blacksholes f64 emulation
1 parent 6ca2075 commit 672a111

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

dpbench/benchmarks/black_scholes/black_scholes_numba_mlir_k.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import numba_mlir.kernel as nb
88

99

10-
@nb.kernel
10+
@nb.kernel(gpu_fp64_truncate="auto")
1111
def _black_scholes_kernel(nopt, price, strike, t, rate, volatility, call, put):
1212
mr = -rate
1313
sig_sig_two = volatility * volatility * 2

dpbench/benchmarks/black_scholes/black_scholes_numba_mlir_n.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ def _nberf(x):
1313
return erf(x)
1414

1515

16-
@nb.njit(parallel=True, fastmath=True)
16+
@nb.njit(parallel=True, fastmath=True, gpu_fp64_truncate="auto")
1717
def _black_scholes(price, strike, t, rate, volatility, call, put):
1818
mr = -rate
1919
sig_sig_two = volatility * volatility * 2

dpbench/benchmarks/black_scholes/black_scholes_numba_mlir_p.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010

1111
# blackscholes implemented as a parallel loop using numba.prange
12-
@nb.njit(parallel=True, fastmath=True)
12+
@nb.njit(parallel=True, fastmath=True, gpu_fp64_truncate="auto")
1313
def _black_scholes(nopt, price, strike, t, rate, volatility, call, put):
1414
mr = -rate
1515
sig_sig_two = volatility * volatility * 2

0 commit comments

Comments
 (0)