batchedgemmソースコード？

Question

私には特別な問題があります。

私はCUDA 4.1を使用して、特にbatchedgemmを使って自分のMacBookで開発したリサーチコードをいくつか持っています。私は今、私が別の機関から貸し出しているgpuのクラスター上で実行しなければなりません。

私の問題は、クラスタにはCUDA 4.0しかインストールされておらず、高速でアップグレードすることには消極的です。

私はどこでもbatchedgemmのソースを入手して4.0で動作するようにコンパイルすることができますか？

私はバッチ乗算を行うために自分のカーネルを書いたことがありますが、ライブラリよりも約10倍遅いオーダーを実行します - 私は自分のつま先ではなく偉大な男性の肩に立たせたいと思います。

Answer 1

実動クラスタをすばやくアップグレードしたくないと私は理解しています。多くのクラスタでは、複数のバージョンのCUDAツールキットが共存できるモジュールシステムを使用しています。しかし、ドライバは、使用中の最新のCUDAをサポートするバージョンにアップグレードする必要があります。これは、ユーザーの生産コードとアプリケーションをテストして回帰や失敗を避ける必要があるため、彼らが嫌な理由です。

CUBLASはオープンソースではないため、別のマシンでコードを開発し、バッチ処理から高速化する場合は、アップグレードする理由として管理者に提示してください。