Function validate_cpu_vs_gpu

dali/test/python/test_gpu_python_function_operator.py:57–60 · view source on GitHub ↗

(gpu_fun, cpu_fun, num_outputs=1)

Source from the content-addressed store, hash-verified

55
56
57	def validate_cpu_vs_gpu(gpu_fun, cpu_fun, num_outputs=1):
58	gpu_pipe = PythonFunctionPipeline(gpu_fun, "gpu", num_outputs)
59	cpu_pipe = PythonFunctionPipeline(cpu_fun, "cpu", num_outputs)
60	test_utils.compare_pipelines(gpu_pipe, cpu_pipe, BATCH_SIZE, ITERS)
61
62
63	def arrays_arithmetic(in1, in2):

test_simple_arithmFunction · 0.85

test_cupy_kernelFunction · 0.85

test_builtin_funcFunction · 0.85

PythonFunctionPipelineClass · 0.85

no test coverage detected