mirror of
https://github.com/smyalygames/FiniteVolumeGPU.git
synced 2025-11-28 00:09:50 +01:00
30 lines
2.0 KiB
Python
30 lines
2.0 KiB
Python
import numpy as np
|
|
import pycuda.driver as cuda
|
|
|
|
from .simulator import BaseMPISimulator
|
|
|
|
|
|
class CudaMPISimulator(BaseMPISimulator):
|
|
def __init__(self, sim, grid):
|
|
super().__init__(sim, grid, self.__create_pagelocked_memory)
|
|
|
|
def __create_pagelocked_memory(self):
|
|
self.in_e = cuda.pagelocked_empty((int(self.nvars), int(self.read_e[3]), int(self.read_e[2])),
|
|
dtype=np.float32) # np.empty((self.nvars, self.read_e[3], self.read_e[2]), dtype=np.float32)
|
|
self.in_w = cuda.pagelocked_empty((int(self.nvars), int(self.read_w[3]), int(self.read_w[2])),
|
|
dtype=np.float32) # np.empty((self.nvars, self.read_w[3], self.read_w[2]), dtype=np.float32)
|
|
self.in_n = cuda.pagelocked_empty((int(self.nvars), int(self.read_n[3]), int(self.read_n[2])),
|
|
dtype=np.float32) # np.empty((self.nvars, self.read_n[3], self.read_n[2]), dtype=np.float32)
|
|
self.in_s = cuda.pagelocked_empty((int(self.nvars), int(self.read_s[3]), int(self.read_s[2])),
|
|
dtype=np.float32) # np.empty((self.nvars, self.read_s[3], self.read_s[2]), dtype=np.float32)
|
|
|
|
# Allocate data for sending
|
|
self.out_e = cuda.pagelocked_empty((int(self.nvars), int(self.read_e[3]), int(self.read_e[2])),
|
|
dtype=np.float32) # np.empty_like(self.in_e)
|
|
self.out_w = cuda.pagelocked_empty((int(self.nvars), int(self.read_w[3]), int(self.read_w[2])),
|
|
dtype=np.float32) # np.empty_like(self.in_w)
|
|
self.out_n = cuda.pagelocked_empty((int(self.nvars), int(self.read_n[3]), int(self.read_n[2])),
|
|
dtype=np.float32) # np.empty_like(self.in_n)
|
|
self.out_s = cuda.pagelocked_empty((int(self.nvars), int(self.read_s[3]), int(self.read_s[2])),
|
|
dtype=np.float32) # np.empty_like(self.in_s)
|