apache-spark - Spark MLlib 单精度分布式矩阵

Question

是否可以在 spark 中创建包含单精度条目的分布式块矩阵？

根据我从文档中收集到的信息，blockmatrix 的 scala/java 实现需要一个 mllib.Matrix 对象，该对象将值保存为双精度值。有没有办法绕过这个限制？

背景：我使用 GPU 来加速 Sparks 分布式矩阵乘法例程，当乘以双精度矩阵而不是单精度矩阵时，我的 GPU 的执行速度要慢 20 倍。