facebookresearch / TensorComprehensions

A domain specific language to express machine learning workloads.
https://facebookresearch.github.io/TensorComprehensions/
Apache License 2.0
1.76k stars 211 forks source link

Ensure types other than float/int32 are supported #98

Open ftynse opened 6 years ago

ftynse commented 6 years ago
nicolasvasilache commented 6 years ago

Great, thanks!

ftynse commented 6 years ago

Actually, memory promotion takes the byte-size from Halide, so it should just work. https://github.com/facebookresearch/TensorComprehensions/blob/f5e3899ddda03c378cc5ce2717e4f28eb92c70d8/src/core/polyhedral/memory_promotion_heuristic.cc#L431

prigoyal commented 6 years ago

yeah and the byte sizes information is also part of DLTensor