This is an archive of the discontinued LLVM Phabricator instance.

Show First 20 Lines • Show All 140 Lines • ▼ Show 20 Lines	orderings = [
ir.AffineMap.get_permutation([1, 0])		ir.AffineMap.get_permutation([1, 0])
]		]
vec_strategy = ['none', 'dense-inner-loop']		vec_strategy = ['none', 'dense-inner-loop']
for level in levels:		for level in levels:
for ordering in orderings:		for ordering in orderings:
for pwidth in [32]:		for pwidth in [32]:
for iwidth in [32]:		for iwidth in [32]:
for vec in vec_strategy:		for vec in vec_strategy:
for e in [True]:		for e in [True]:
vl = 1 if vec == 0 else 16		vl = 1 if vec == 0 else 16
attr = st.EncodingAttr.get(level, ordering, pwidth, iwidth)		attr = st.EncodingAttr.get(level, ordering, pwidth, iwidth)
opt = (f'parallelization-strategy=none '		opt = (f'parallelization-strategy=none '
f'vectorization-strategy={vec} '		f'vectorization-strategy={vec} '
f'vl={vl} enable-simd-index32={e}')		f'vl={vl} enable-simd-index32={e}')
compiler = sparse_compiler.SparseCompiler(		compiler = sparse_compiler.SparseCompiler(
options=opt, opt_level=0, shared_libs=[support_lib])		options=opt, opt_level=0, shared_libs=[support_lib])
build_compile_and_run_SDDMMM(attr, compiler)		build_compile_and_run_SDDMMM(attr, compiler)
count = count + 1		count = count + 1
# CHECK: Passed 16 tests		# CHECK: Passed 16 tests
print('Passed ', count, 'tests')		print('Passed ', count, 'tests')


if __name__ == '__main__':		if __name__ == '__main__':
main()		main()

[mlir][sparse] fix python indentation in testClosedPublic