replace .cu with cu.hip

This commit is contained in:
Hicham Agueny 2024-02-20 14:25:56 +01:00 committed by GitHub
parent e2b56c6ab9
commit b53f5bf601
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -108,7 +108,7 @@ class EE2D_KP07_dimsplit (BaseSimulator):
#self.kernel = module.get_function("KP07DimsplitKernel") #self.kernel = module.get_function("KP07DimsplitKernel")
#self.kernel.prepare("iiffffffiiPiPiPiPiPiPiPiPiPiiii") #self.kernel.prepare("iiffffffiiPiPiPiPiPiPiPiPiPiiii")
# #
kernel_file_path = os.path.abspath(os.path.join('cuda', 'EE2D_KP07_dimsplit.cu')) kernel_file_path = os.path.abspath(os.path.join('cuda', 'EE2D_KP07_dimsplit.cu.hip'))
with open(kernel_file_path, 'r') as file: with open(kernel_file_path, 'r') as file:
kernel_source = file.read() kernel_source = file.read()