diff options
Diffstat (limited to 'ptx/tools/implicit_ld_src.py')
| -rw-r--r-- | ptx/tools/implicit_ld_src.py | 31 |
1 files changed, 0 insertions, 31 deletions
diff --git a/ptx/tools/implicit_ld_src.py b/ptx/tools/implicit_ld_src.py deleted file mode 100644 index cb6780e..0000000 --- a/ptx/tools/implicit_ld_src.py +++ /dev/null @@ -1,31 +0,0 @@ -import os
-import subprocess
-import tempfile
-
-types = ["b8", "b16", "b32", "b64", "u8", "u16", "u32", "u64", "s8", "s16", "s32", "s64", "f32", "f64"]
-
-for input_type in types:
- for op_type in types:
- with tempfile.TemporaryDirectory() as dir:
- f_name = os.path.join(dir, 'ptx')
- out_name = os.path.join(dir, 'out')
- with open(f_name, 'w') as f:
- f.write(
- f"""
- .version 6.5
- .target sm_30
- .address_size 64
- .visible .entry VecAdd_kernel(
- .param .{input_type} input
- )
- {{
- .reg.{op_type} r1;
- ld.param.{op_type} r1, [input];
- ret;
- }}
- """)
- err = subprocess.run(f"ptxas {f_name} -o {out_name}")
- if err.returncode == 0:
- print(f"{op_type} {input_type}")
- else:
- print(f"[INVALID] {op_type} {input_type}")
\ No newline at end of file |