Skip to content
Snippets Groups Projects

Adding Dim3 class

Merged Christoph Alt requested to merge ob28imeq/pystencils-sfg:gpu_extensions into master
Viewing commit 80afd8c1
Prev
Show latest version
2 files
+ 5
4
Preferences
Compare changes
Files
2
@@ -4,8 +4,9 @@ from pystencilssfg import lang
def dim3class(gpu_runtime_header: str, *, cls_name: str = "dim3"):
"""
>>> dim3 = dim3class("<hip/hip_runtime.h>")
>>> dim3.ctor(64, 1, 1)
'dim3{64, 1, 1}'
>>> dim3().ctor(64, 1, 1)
dim3{64, 1, 1}
Args:
gpu_runtime_header: String with the name of the gpu runtime header
cls_name: String with the acutal name (default "dim3")