[DRAFT] Train generalized machine learning model using pointwise data set.

[rdspring1]

This PR creates a training script using the dataset generated by the script in https://github.com/NVIDIA/Fuser/pull/3255.

Overview

Decision tree is trained on a set of add and exp pointwise fusions. Empirical test is gelu(a + b).

Inputs for decision tree is:

• sum total bytes for input tensors
• sum total bytes for output tensors
• vectorize factor
• unroll factor

Result