Skip to content

Commit d8da811

Browse files
committed
Fix select_chunk calls in GPU implementation and syntax error
1 parent f066728 commit d8da811

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/fftvis/gpu/gpu_simulate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -416,7 +416,7 @@ def _evaluate_vis_chunk(
416416

417417
for source_chunk_idx in range(n_source_chunks):
418418
# Select a chunk of sources above the horizon
419-
topo, flux_sqrt, nsim_sources = coord_mgr.select_chunk(source_chunk_idx)
419+
topo, flux_sqrt, nsim_sources = coord_mgr.select_chunk(source_chunk_idx, time_total_idx)
420420

421421
# Ensure topo and flux_sqrt are on GPU
422422
if not isinstance(topo, cp.ndarray):

tests/test_performance_cpu_vs_gpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -392,4 +392,4 @@ def test_scaling_analysis(self):
392392
if len(sys.argv) > 1:
393393
pytest.main([__file__, f"::TestPerformanceCPUvsGPU::{sys.argv[1]}", "-v", "-s"])
394394
else:
395-
pytest.main([__file__, "-v", "-s"])f
395+
pytest.main([__file__, "-v", "-s"])

0 commit comments

Comments
 (0)