Commit8b01c59

authored

added in dask-cupy convolve_2d test (#823)

1 parentb2b71fd commit8b01c59Copy full SHA for 8b01c59

File tree

-3

lines changed

-3

lines changed

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -373,6 +373,7 @@ def _convolve_2d_cupy(data, kernel):`
`373`	`373`	`_convolve_2d_cuda[griddim,blockdim](data,kernel,cupy.asarray(out))`
`374`	`374`	`returnout`
`375`	`375`
	`376`	`+`
`376`	`377`	`def_convolve_2d_dask_cupy(data,kernel):`
`377`	`378`	`data=data.astype(cupy.float32)`
`378`	`379`	`pad_h=kernel.shape[0]//2`

Lines changed: 5 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -264,9 +264,11 @@ def test_2d_convolution_gpu(`
`264`	`264`	`dask_cupy_agg=xr.DataArray(`
`265`	`265`	`da.from_array(cupy.asarray(convolve_2d_data),chunks=(3,3))`
`266`	`266`	`)`
`267`		`-withpytest.raises(NotImplementedError)ase_info:`
`268`		`-convolution_2d(dask_cupy_agg,kernel_custom)`
`269`		`-asserte_info`
	`267`	`+result_kernel_annulus=convolve_2d(dask_cupy_agg.data,kernel_annulus_2_2_2_1)`
	`268`	`+assertisinstance(result_kernel_annulus,da.Array)`
	`269`	`+np.testing.assert_allclose(`
	`270`	`+result_kernel_annulus.compute().get(),convolution_kernel_annulus_2_2_1,equal_nan=True`
	`271`	`+ )`
`270`	`272`
`271`	`273`
`272`	`274`	`deftest_calc_cellsize_unit_input_attrs(convolve_2d_data):`

Comments

(0)