Use Host model for USMNdArray

IntelPython · Jan 3, 2024 · dcef554 · dcef554
1 parent 2b94982
commit dcef554
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numba_dpex/core/datamodel/models.py b/numba_dpex/core/datamodel/models.py
@@ -291,7 +291,7 @@ def _init_data_model_manager() -> datamodel.DataModelManager:
 
 # Register the USMNdArray type to USMArrayDeviceModel in numba's default data
 # model manager
-register_model(USMNdArray)(USMArrayDeviceModel)
+register_model(USMNdArray)(USMArrayHostModel)
 
 # Register the DpnpNdArray type to USMArrayHostModel in numba's default data
 # model manager