5 Commits

Author	SHA1	Message	Date
Bartłomiej Kocot	ced5af16f7	Extend support for contraction 6D (#1207) ... * Extend support for contraction up to 5D * Extend contraction bilinear instances * Fix interface test * Add 6d support, remove 3d,4d,5d * Fixes * Fix readme * Make defualt dim for contraction instances	2024-04-09 23:46:21 +02:00
Bartlomiej Wroblewski	4ef704d8a6	Add support for mixed precision in contraction scale and bilinear (#973) ... * Add support for mixed precision in contraction scale and bilinear (#936) * Extract common functionality to separate files * Reference contraction: Remove incorrect consts from type_converts * Reference contraction: Add missing type_convert for dst value * Reference contraction: Fix incorrect order of B matrix dimensions * Add support for mixed precision in contraction scale and bilinear * Move using statements from instances to a common file * Move using statements from examples to a common file * Fix the order of B matrix dimensions across examples and profiler * Fix the computation of error threshold * Make ComputeDataType an optional argument * Include possible DataType -> ComputeDataType casting error in the threshold * Remove commented code * Make the ComputeDataType an optional argument in instance --------- Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>	2023-11-02 14:26:33 -07:00
Illia Silin	4daedf8ca5	Revert "Add support for mixed precision in contraction scale and bilinear" (#967) ... * Revert "Add support for mixed precision in contraction scale and bilinear (#936)" This reverts commit f07485060e. * revert commits #957 and #960	2023-10-05 14:58:23 -07:00
Bartlomiej Wroblewski	f07485060e	Add support for mixed precision in contraction scale and bilinear (#936) ... * Extract common functionality to separate files * Reference contraction: Remove incorrect consts from type_converts * Reference contraction: Add missing type_convert for dst value * Reference contraction: Fix incorrect order of B matrix dimensions * Add support for mixed precision in contraction scale and bilinear * Move using statements from instances to a common file * Move using statements from examples to a common file * Fix the order of B matrix dimensions across examples and profiler * Fix the computation of error threshold * Make ComputeDataType an optional argument * Include possible DataType -> ComputeDataType casting error in the threshold * Remove commented code	2023-09-29 10:54:31 -05:00
Bartłomiej Kocot	642d5e9155	Add contraction profiler and tests (#701) ... * Add contraction profiler and tests * Build and style fixes * Allow to use any elementwise operator for ref_contraction * Introduce profile_contraction_scale and profile_contraction_bilinear * Make ref_contraction generic and extend interface tests * Stylistic minor fixes * Extend test_contraction_interface	2023-05-15 09:46:52 -05:00