def main(): dataset_size = 100000 src_dtype = np.float64 dst_dtype = np.uint64 nbits = 20 dataset = make_vec_dataset(dataset_size, dtype=src_dtype) mt.trace( vector.encode_vec_to_uint, target_args=dict(v=dataset, dtype=dst_dtype, nbits=nbits), related_traces_output_mode=mt.RelatedTracesOutputMode.FOR_EACH_FILE, include_patterns={'*/fpq/*'}, exclude_patterns=None)
def main(): dataset_size = 100000 src_dtype = np.float64 dst_dtype = np.uint64 nbits = 20 dataset = fp.encode_fp_to_std_unorm(x=make_unorm_dataset(dataset_size, dtype=src_dtype), dtype=dst_dtype, nbits=nbits) mt.trace( fp.decode_std_unorm_to_fp, target_args=dict(x=dataset, dtype=src_dtype, nbits=nbits), related_traces_output_mode=mt.RelatedTracesOutputMode.FOR_EACH_FILE, include_patterns={'*/fpq/*'}, exclude_patterns=None)
def main(): dataset_size = 100000 src_dtype = np.float64 dst_dtype = np.uint64 dataset = quaternion.encode_quat_to_uint(q=make_quat_dataset( dataset_size, dtype=src_dtype), dtype=dst_dtype) mt.trace( quaternion.decode_uint_to_quat, target_args=dict( q=dataset, dtype=src_dtype, ), related_traces_output_mode=mt.RelatedTracesOutputMode.FOR_EACH_FILE, include_patterns={'*/fpq/*'}, exclude_patterns=None)