Skip to content

Commit 2540627

Browse files
update example
1 parent b254862 commit 2540627

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/cuda/going_green_performance_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
raise e
3434

3535
from kernel_tuner.energy import energy
36-
from kernel_tuner.nvml import get_nvml_gr_clocks
36+
from kernel_tuner.observers.nvml import get_nvml_gr_clocks
3737

3838
def get_default_parser():
3939
parser = argparse.ArgumentParser(
@@ -60,7 +60,7 @@ def get_default_parser():
6060
args = parser.parse_args()
6161

6262
ridge_frequency, freqs, nvml_power, fitted_params, scaling = energy.create_power_frequency_model(device=args.device,
63-
n_samples=args.samples,
63+
n_samples=int(args.samples),
6464
verbose=True,
6565
nvidia_smi_fallback=args.nvidia_smi_fallback,
6666
use_locked_clocks=args.locked_clocks)

0 commit comments

Comments
 (0)