Function per_iter_shape_pipeline

dali/test/python/test_external_source_parallel.py:1130–1150

(
    shapes, py_num_workers=4, batch_size=4, parallel=True, bytes_per_sample_hint=None
)

def per_iter_shape_pipeline(
    shapes, py_num_workers=4, batch_size=4, parallel=True, bytes_per_sample_hint=None
):
    @dali.pipeline_def
    def pipeline():
        return dali.fn.external_source(
            PerIterShapeSource(shapes),
            batch=False,
            parallel=parallel,
            bytes_per_sample_hint=bytes_per_sample_hint,
        )

    pipe = pipeline(
        batch_size=batch_size,
        py_num_workers=py_num_workers,
        device_id=0,
        num_threads=4,
        py_start_method="spawn",
    )
    pipe.build()
    return pipe


def test_no_parallel_no_shm():