Home
last modified time | relevance | path

Searched refs:input_type (Results 1 – 5 of 5) sorted by relevance

/llvm-project/mlir/test/Integration/GPU/CUDA/sm90/python/
H A Dmatmul.py90 input_type=np.float16, argument
106 input_type,
118 input_type,
156 input_type=np.float16, argument
173 ity = "f16" if input_type == np.float16 else "f32"
204 input_type,
219 a = np.random.randn(M, K).astype(input_type)
220 b = np.random.randn(K, N).astype(input_type)
225 input_type,
248 ref = a.astype(input_type) @ b.astype(input_type)
/llvm-project/llvm/utils/mlgo-utils/mlgo/corpus/
H A Dextract_ir.py44 "--input_type",
137 elif args.input_type == "json":
142 elif args.input_type == "params":
153 elif args.input_type == "directory":
161 elif args.input_type == "bazel_aquery":
167 logging.error("Unknown input type: %s", args.input_type)
/llvm-project/mlir/test/Integration/GPU/CUDA/sm90/python/tools/
H A DmatmulBuilder.py132 input_type=np.float16, argument
164 input_type=np.float16, argument
175 assert input_type == np.float16
186 a_elem_ty = get_mlir_ty(input_type)
187 b_elem_ty = get_mlir_ty(input_type)
239 input_type, output_type, M, N, K, BLOCK_M, BLOCK_N, BLOCK_K, num_stages, True
657 input_type=np.float16,
668 assert input_type == np.float16
679 a_elem_ty = get_mlir_ty(input_type)
680 b_elem_ty = get_mlir_ty(input_type)
664 generate_matmul_multistage( input_type=np.float16, output_type=np.float32, M=4096, N=4096, K=4096, BLOCK_M=128, BLOCK_N=128, BLOCK_K=64, num_stages=3, ) global() argument
[all...]
/llvm-project/pstl/test/std/numerics/numeric.ops/
H A Dscan.pass.cpp80 typedef typename std::iterator_traits<InputIterator>::value_type input_type; in inclusive_scan_serial() typedef
81 return inclusive_scan_serial(first, last, result, std::plus<input_type>()); in inclusive_scan_serial()
/llvm-project/llvm/utils/mlgo-utils/tests/corpus/
H A Dextract_ir_script.test10 # RUN: %python %scripts_dir/corpus/extract_ir.py --input=%t.dir --input_type=directory --output_dir…