fix sz.sh for llvm
@@ -86,10 +86,10 @@ class LLVM:
# TODO: this makes compile times so much faster
if getenv("LLVMOPT"):
- llvm.set_option('', '-force-vector-interleave=4') # this makes sum the same speed as torch, it also doubles the (slow) conv speed
+ llvm.set_option(str(), '-force-vector-interleave=4') # this makes sum the same speed as torch, it also doubles the (slow) conv speed
if DEBUG >= 4:
- llvm.set_option('', '--debug-only=loop-vectorize')
- #llvm.set_option('', '--debug')
+ llvm.set_option(str(), '--debug-only=loop-vectorize')
+ #llvm.set_option(str(), '--debug')
# does this do anything?
builder = llvm.create_pass_manager_builder()
@@ -100,7 +100,7 @@ class LLVM:
builder.populate(LLVM.optimizer)
LLVM.target_machine.set_asm_verbosity(True)
- backing_mod = llvm.parse_assembly("")
+ backing_mod = llvm.parse_assembly(str())
backing_mod.triple = llvm.get_process_triple()
LLVM.engine = llvm.create_mcjit_compiler(backing_mod, LLVM.target_machine)
@@ -236,10 +236,6 @@ class LLVMBuffer(ExplicitExecAST):
lambda shape: (shape[0], shape[1]//DY, DY, shape[2], shape[3]//DX, DX, shape[4], shape[5], shape[6]),
(0,1,3,4,6,7,8,2,5))
kernel_output_axis = [-2, -1]
-
- if DEBUG >= 2:
- print("new:", k.shapes)
- print("new:", k.strides)
"""
# the 4x4 need to go all the way at the end, even after reduce
