replace .cu with cu.hip

This commit is contained in:
Hicham Agueny
2024-02-20 14:26:26 +01:00
committed by GitHub
parent b53f5bf601
commit 56445c201e

View File

@@ -88,7 +88,7 @@ class FORCE (Simulator.BaseSimulator):
self.g = np.float32(g)
#Get kernels
# module = context.get_module("cuda/SWE2D_FORCE.cu",
# module = context.get_module("cuda/SWE2D_FORCE.cu.hip",
# defines={
# 'BLOCK_WIDTH': self.block_size[0],
# 'BLOCK_HEIGHT': self.block_size[1]