File "analyse_elastic.py", line 78, in
dataset = xr.open_dataset(r'/home/tvo/Data_Disk/mumax-python-server/final_nc_files/11-03-2024_20-46-57_ndSweep_C11_C12_C44_asigma_etaV_elastic_simulation.nc').isel(C11=0, C12=0, C44=1, etaV=0, comp=0, wavetype=2, asigma=slice(0,2)).fft_cellwise(FFT_dims=['y', 't'], multiple_GPUs=True, chunks={'t': 500, 'x': 128, 'y': 128})
File "/home/tvo/.local/lib/python3.8/site-packages/cuxrft/xr_controller_fft.py", line 559, in fft_cellwise
out = dask.array.rechunk(dask.array.squeeze(sel_chunk(data, chunks, out, data_var, fftAxis, "forward", useGPU=GPUs, multiple_GPUs=multiple_GPUs)), rechunkDict)
File "/home/tvo/.local/lib/python3.8/site-packages/cuxrft/xr_controller_fft.py", line 297, in sel_chunk
out[sc] = sel_chunk(data.isel({key: slice(i, i+chunkLength)}), chunks, out, resultDim, fftAxis, fftDirection, fftNorm, chunkedDims, useGPU=useGPU, sc=sc, multiple_GPUs=multiple_GPUs)
File "/home/tvo/.local/lib/python3.8/site-packages/dask/array/core.py", line 1931, in setitem
dsk = setitem_array(out, self, key, value)
File "/home/tvo/.local/lib/python3.8/site-packages/dask/array/slicing.py", line 1868, in setitem_array
raise ValueError(
ValueError: shape mismatch: value array of shape (256, 1001, 2, 256) could not be broadcast to indexing result of shape (256, 1001, 2, 128)
For
dataset = xr.open_dataset(r'/home/tvo/Data_Disk/mumax-python-server/final_nc_files/11-03-2024_20-46-57_ndSweep_C11_C12_C44_asigma_etaV_elastic_simulation.nc').isel(C11=0, C12=0, C44=1, etaV=0, comp=0, wavetype=2, asigma=slice(0,2)).fft_cellwise(FFT_dims=['y', 't'], multiple_GPUs=True, chunks={'t': 500, 'x': 128, 'y': 128})
File "analyse_elastic.py", line 78, in
dataset = xr.open_dataset(r'/home/tvo/Data_Disk/mumax-python-server/final_nc_files/11-03-2024_20-46-57_ndSweep_C11_C12_C44_asigma_etaV_elastic_simulation.nc').isel(C11=0, C12=0, C44=1, etaV=0, comp=0, wavetype=2, asigma=slice(0,2)).fft_cellwise(FFT_dims=['y', 't'], multiple_GPUs=True, chunks={'t': 500, 'x': 128, 'y': 128})
File "/home/tvo/.local/lib/python3.8/site-packages/cuxrft/xr_controller_fft.py", line 559, in fft_cellwise
out = dask.array.rechunk(dask.array.squeeze(sel_chunk(data, chunks, out, data_var, fftAxis, "forward", useGPU=GPUs, multiple_GPUs=multiple_GPUs)), rechunkDict)
File "/home/tvo/.local/lib/python3.8/site-packages/cuxrft/xr_controller_fft.py", line 297, in sel_chunk
out[sc] = sel_chunk(data.isel({key: slice(i, i+chunkLength)}), chunks, out, resultDim, fftAxis, fftDirection, fftNorm, chunkedDims, useGPU=useGPU, sc=sc, multiple_GPUs=multiple_GPUs)
File "/home/tvo/.local/lib/python3.8/site-packages/dask/array/core.py", line 1931, in setitem
dsk = setitem_array(out, self, key, value)
File "/home/tvo/.local/lib/python3.8/site-packages/dask/array/slicing.py", line 1868, in setitem_array
raise ValueError(
ValueError: shape mismatch: value array of shape (256, 1001, 2, 256) could not be broadcast to indexing result of shape (256, 1001, 2, 128)
For
dataset = xr.open_dataset(r'/home/tvo/Data_Disk/mumax-python-server/final_nc_files/11-03-2024_20-46-57_ndSweep_C11_C12_C44_asigma_etaV_elastic_simulation.nc').isel(C11=0, C12=0, C44=1, etaV=0, comp=0, wavetype=2, asigma=slice(0,2)).fft_cellwise(FFT_dims=['y', 't'], multiple_GPUs=True, chunks={'t': 500, 'x': 128, 'y': 128})