Оптимизация мутации массива в Clojure

Question

Прочитав несколько постов в блоге на эту тему, я обнаружил, что мутирует массив в Clojure так:

(defn m [xs ys] 
    (dotimes [i (count xs)] 
    (aset #^ints ys (int i) 
    (int (* (int 3) (int (aget #^ints xs (int i)))))))) 

где (def xs (into-array Integer/TYPE (range 1000000))) и (def ys (into-array Integer/TYPE (range 1000000)))

уходило в среднем 14ms согласно Критерию, тогда как Java должен делать то же самое,

public static int[] m(int[] x, int[] y) 
{ 
    for(int i=0; i<x.length; i++) 
    y[i] = 3*x[i]; 
    return y; 
} 

занимает в среднем 800US. **

Я делаю все, что в моих силах, чтобы все ускорилось, и есть ли еще какие-нибудь пути оптимизации?

** Я приуроченная их с помощью критериум с (report-result (bench (m xs ys)) :verbose) и (report-result (bench (. Test m xs ys)) :verbose)

Answer 1

Попробуйте это на Clojure 1.3:

(set! *unchecked-math* true) 

(defn m [^longs xs ^longs ys] 
    (dotimes [i (count xs)] 
    (aset ys i 
     (* 3 (aget xs i))))) 

Answer 2

Если вы хотите скорость, вы должны войти в мир примитивов, а не оставлять его, пока вы не закончите. Это бесполезно, начиная с коробкой Integer i, а затем преобразуя ее в примитив на каждом сайте использования. Может быть, вы можете сделать dotimes произвести ints (тип-hint объявление i), но не уверен. То, что я знаю, является конструкцией loop-recur с примитивными инициализаторами циклов: (loop [i (int 0)] ... (recur (unchecked-inc i)). Кроме того, в вашем примере у вас есть (int 3). Вам нужно let, что заранее, чтобы не повторять распаковку на каждой итерации.

BTW, вы можете использовать (int-array (range 1000000)) для создания вашего инициализированного массива и всего лишь (int-array 1000000) для пустого.

UPDATE

По состоянию на Clojure 1.3 с расширенной поддержкой примитивов, большинство из того, что я написал выше, не применяется больше. dotimes уже использует примитивную арифметику, поэтому все, что вам нужно написать, чтобы получить полную производительность является

(dotimes [i (alength ^ints xs)] 
    (aset ^ints ys i (unchecked-multiply (aget ^ints xs i) 3) 

В принципе, нет int Конструкторов необходимо, и использовать unchecked-multiply.